Balanced masking strategy for multi-label image classification

doi:10.1016/j.neucom.2022.11.022

	Balanced masking strategy for multi-label image classification
	Yuan, Jin 1; Zhang, Yao 2; Shi, Zhongchao 3; Geng, Xin 1; Fan, Jianping 3; Rui, Yong 3
	2023-02-14
发表期刊	NEUROCOMPUTING
ISSN	0925-2312
卷号	522 页码:64-72
摘要	Data imbalance is an essential issue in multi-label image classification that may reduce the model's gen-eralization ability. Unlike oversampling or undersampling in single-label image classification, simply removing or repeating some images in multi-label classification will influence labels in other categories due to complex label dependencies. Current methods usually tackle this issue from either data resam-pling or loss function design. This paper proposes a balanced masking strategy to solve the imbalance problem in feature space for graph-based methods. By selectively manipulating the label node embed-ding of the graph, the model can gain the priors of the majority of samples to maintain intra-class and inter-class balance simultaneously, thus performing better and more stable. Extensive experiments verify the adaptability and effectiveness of our method, which can be easily applied to any graph-based multi -label model to achieve highly competitive performance.(c) 2022 Published by Elsevier B.V.
关键词	Multi -label image classification Data imbalance Masking strategy Graph neural network
DOI	10.1016/j.neucom.2022.11.022
收录类别	SCI
语种	英语
WOS研究方向	Computer Science
WOS类目	Computer Science, Artificial Intelligence
WOS记录号	WOS:000904832400006
出版者	ELSEVIER
引用统计	被引频次：5[WOS] [WOS记录] [WOS相关记录]
文献类型	期刊论文
条目标识符	http://119.78.100.204/handle/2XEOYT63/20122
专题	中国科学院计算技术研究所期刊论文
通讯作者	Geng, Xin; Rui, Yong
作者单位	1.Southeast Univ, Nanjing, Peoples R China 2.Chinese Acad Sci, Inst Comp Technol, Beijing, Peoples R China 3.Lenovo Res, Beijing, Peoples R China
推荐引用方式 GB/T 7714	Yuan, Jin,Zhang, Yao,Shi, Zhongchao,et al. Balanced masking strategy for multi-label image classification[J]. NEUROCOMPUTING,2023,522:64-72.
APA	Yuan, Jin,Zhang, Yao,Shi, Zhongchao,Geng, Xin,Fan, Jianping,&Rui, Yong.(2023).Balanced masking strategy for multi-label image classification.NEUROCOMPUTING,522,64-72.
MLA	Yuan, Jin,et al."Balanced masking strategy for multi-label image classification".NEUROCOMPUTING 522(2023):64-72.