arrow
返回

SFPL: Sample-specific fine-grained prototype learning for imbalanced medical image classification

delete2024-10-01
delete0
delete
OA
AI
Y
Yongbei Zhu
W
Wang Shuo *
Y
Yu He
W
Weimin Li
J
Jie Tian
DOI:10.1016/j.media.2024.103281delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Imbalanced classification is a common and difficult task in many medical image analysis applications. However, most existing approaches focus on balancing feature distribution and classifier weights between classes, while ignoring the inner-class heterogeneity and the individuality of each sample. In this paper, we proposed a sample-specific fine-grained prototype learning (SFPL) method to learn the fine-grained representation of the majority class and learn a cosine classifier specifically for each sample such that the classification model is highly tuned to the individual's characteristic. SFPL first builds multiple prototypes to represent the majority class, and then updates the prototypes through a mixture weighting strategy. Moreover, we proposed a uniform loss based on set representations to make the fine-grained prototypes distribute uniformly. To establish associations between fine-grained prototypes and cosine classifier, we propose a selective attention aggregation module to select the effective fine-grained prototypes for final classification. Extensive experiments on three different tasks demonstrate that SFPL outperforms the state-of-the-art (SOTA) methods. Importantly, as the imbalance ratio increases from 10 to 100, the improvement of SFPL over SOTA methods increases from 2.2% to 2.4%; as the training data decreases from 800 to 100, the improvement of SFPL over SOTA methods increases from 2.2% to 3.8%.
Keyword:
Imbalanced classification
Contrastive learning
Fine-grained prototype
Sample-specific classifier

期刊

Medical Image Analysis 封面图
Medical Image Analysis
IF:
11.8
论文数:
3.9K
被引数:
2.4W

机构

B
Beihang University
学者数:
5.2W
论文数: 4.1W
被引数: 37
I
institute of automation, cas
学者数:
2.2K
论文数: 2.1K
被引数: 2
引用论文

引用论文

The Lung Image Database Consortium, (LIDC) and Image Database Resource Initiative (IDRI): A Completed Reference Database of Lung Nodules on CT Scans肺图像数据库联盟 (LIDC) 和图像数据库资源倡议 (IDRI): ct扫描上完整的肺结节参考数据库
err2011-01-24
err1.8K
errOAAI
errArmato, Samuel G., III; McLennan, Geoffrey; Bidaut, Luc; McNitt-Gray, Michael F.; Meyer, Charles R.; Reeves, Anthony P.; Zhao, Binsheng; Aberle, Denise R.; Henschke, Claudia I.; Hoffman, Eric A.; Kazerooni, Ella A.; MacMahon, Heber; van Beek, Edwin J. R.; Yankelevitz, David; Biancardi, Alberto M.; Bland, Peyton H.; Brown, Matthew S.; Engelmann, Roger M.; Laderach, Gary E.; Max, Daniel; Pais, Richard C.; Qing, David P-Y; Roberts, Rachael Y.; Smith, Amanda R.; Starkey, Adam; Batra, Poonam; Caligiuri, Philip; Farooqi, Ali; Gladish, Gregory W.; Jude, C. Matilda; Munden, Reginald F.; Petkovska, Iva; Quint, Leslie E.; Schwartz, Lawrence H.; Sundaram, Baskaran; Dodd, Lori E.; Fenimore, Charles; Gur, David; Petrick, Nicholas; Freymann, John; Kirby, Justin; Hughes, Brian; Casteele, Alessi Vande; Gupte, Sangeeta; Sallam, Maha; Heath, Michael D.; Kuhn, Michael H.; Dharaiya, Ekta; Burns, Richard; Fryd, David S.; Salganicoff, Marcos; Anand, Vikram; Shreter, Uri; Vastagh, Stephen; Croft, Barbara Y.; Clarke, Laurence P.
err分享
err收藏
Introduction
err2005-08-25
err0
PREAI
errMary Coleman; Catalina Betancur
err分享
err收藏
Subchronic Toxicity of Copper Oxide Nanoparticles and Its Attenuation with the Help of a Combination of Bioprotectors
err2014-07-14
err0
errOAAI
errLarisa Privalova; Boris Katsnelson; Nadezhda Loginova; Vladimir Gurvich; Vladimir Shur; Irene Valamina; Oleg Makeyev; Marina Sutunkova; Ilzira Minigalieva; Ekaterina Kireyeva; Vadim Rusakov; Anastasia Tyurnina; Roman Kozin; Ekaterina Meshtcheryakova; Artem Korotkov; Eugene Shuman; Anastasia Zvereva; Svetlana Kostykova
err分享
err收藏
Arsenic removal from aqueous solutions by adsorption using novel MIL-53(Fe) as a highly efficient adsorbent使用新型MIL-53(Fe) 作为高效吸附剂通过吸附从水溶液中去除砷
err2015-01-01
err0
PREAI
errTuan. A. Vu; Giang. H. Le; Canh. D. Dao; Lan. Q. Dang; Kien. T. Nguyen; Quang. K. Nguyen; Phuong. T. Dang; Hoa. T. K. Tran; Quang. T. Duong; Tuyen. V. Nguyen; Gun. D. Lee
err分享
err收藏
Finite Mixture Models有限混合模型
err2019-03-07
err559
PREAI
errMcLachlan, Geoffrey J.; Lee, Sharon X.; Rathnayake, Suren I.
err分享
err收藏
学者 查看更多内容