Medical-Knowledge Driven Multiple Instance Learning for Classifying Severe Abdominal Anomalies on Prenatal Ultrasound

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liang, Huanwen, Xu, Jingxian, Zhang, Yuanji, Huang, Yuhao, Zhang, Yuhan, Yang, Xin, Li, Ran, Deng, Xuedong, Liu, Yanjun, Tao, Guowei, Wu, Yun, Zhao, Sheng, Gao, Xinru, Ni, Dong
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918079325274112
author Liang, Huanwen
Xu, Jingxian
Zhang, Yuanji
Huang, Yuhao
Zhang, Yuhan
Yang, Xin
Li, Ran
Deng, Xuedong
Liu, Yanjun
Tao, Guowei
Wu, Yun
Zhao, Sheng
Gao, Xinru
Ni, Dong
author_facet Liang, Huanwen
Xu, Jingxian
Zhang, Yuanji
Huang, Yuhao
Zhang, Yuhan
Yang, Xin
Li, Ran
Deng, Xuedong
Liu, Yanjun
Tao, Guowei
Wu, Yun
Zhao, Sheng
Gao, Xinru
Ni, Dong
contents Fetal abdominal malformations are serious congenital anomalies that require accurate diagnosis to guide pregnancy management and reduce mortality. Although AI has demonstrated significant potential in medical diagnosis, its application to prenatal abdominal anomalies remains limited. Most existing studies focus on image-level classification and rely on standard plane localization, placing less emphasis on case-level diagnosis. In this paper, we develop a case-level multiple instance learning (MIL)-based method, free of standard plane localization, for classifying fetal abdominal anomalies in prenatal ultrasound. Our contribution is three-fold. First, we adopt a mixture-of-attention-experts module (MoAE) to weight different attention heads for various planes. Secondly, we propose a medical-knowledge-driven feature selection module (MFS) to align image features with medical knowledge, performing self-supervised image token selection at the case-level. Finally, we propose a prompt-based prototype learning (PPL) to enhance the MFS. Extensively validated on a large prenatal abdominal ultrasound dataset containing 2,419 cases, with a total of 24,748 images and 6 categories, our proposed method outperforms the state-of-the-art competitors. Codes are available at:https://github.com/LL-AC/AAcls.
format Preprint
id arxiv_https___arxiv_org_abs_2507_01401
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Medical-Knowledge Driven Multiple Instance Learning for Classifying Severe Abdominal Anomalies on Prenatal Ultrasound
Liang, Huanwen
Xu, Jingxian
Zhang, Yuanji
Huang, Yuhao
Zhang, Yuhan
Yang, Xin
Li, Ran
Deng, Xuedong
Liu, Yanjun
Tao, Guowei
Wu, Yun
Zhao, Sheng
Gao, Xinru
Ni, Dong
Computer Vision and Pattern Recognition
Artificial Intelligence
Fetal abdominal malformations are serious congenital anomalies that require accurate diagnosis to guide pregnancy management and reduce mortality. Although AI has demonstrated significant potential in medical diagnosis, its application to prenatal abdominal anomalies remains limited. Most existing studies focus on image-level classification and rely on standard plane localization, placing less emphasis on case-level diagnosis. In this paper, we develop a case-level multiple instance learning (MIL)-based method, free of standard plane localization, for classifying fetal abdominal anomalies in prenatal ultrasound. Our contribution is three-fold. First, we adopt a mixture-of-attention-experts module (MoAE) to weight different attention heads for various planes. Secondly, we propose a medical-knowledge-driven feature selection module (MFS) to align image features with medical knowledge, performing self-supervised image token selection at the case-level. Finally, we propose a prompt-based prototype learning (PPL) to enhance the MFS. Extensively validated on a large prenatal abdominal ultrasound dataset containing 2,419 cases, with a total of 24,748 images and 6 categories, our proposed method outperforms the state-of-the-art competitors. Codes are available at:https://github.com/LL-AC/AAcls.
title Medical-Knowledge Driven Multiple Instance Learning for Classifying Severe Abdominal Anomalies on Prenatal Ultrasound
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2507.01401