中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
首页 机构 成果 学者
  
  1. CAS IR Grid
  2. 计算技术研究所
  3. 中国科学院计算技术研究所
  4. 中国科学院计算技术研究所期刊论文
Balanced masking strategy for multi-label image classification

文献类型：期刊论文

作者Yuan, Jin3; Zhang, Yao2; Shi, Zhongchao1; Geng, Xin3; Fan, Jianping1; Rui, Yong1
刊名NEUROCOMPUTING
出版日期2023-02-14
卷号522页码:64-72
ISSN号0925-2312
关键词Multi -label image classification Data imbalance Masking strategy Graph neural network
DOI10.1016/j.neucom.2022.11.022
英文摘要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.
WOS研究方向Computer Science
语种英语
出版者ELSEVIER
WOS记录号WOS:000904832400006
源URL[http://119.78.100.204/handle/2XEOYT63/20121]  
专题中国科学院计算技术研究所期刊论文
通讯作者Geng, Xin; Rui, Yong
作者单位1.Lenovo Res, Beijing, Peoples R China
2.Chinese Acad Sci, Inst Comp Technol, Beijing, Peoples R China
3.Southeast Univ, Nanjing, 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.

入库方式： OAI收割

来源：计算技术研究所

浏览0
下载0
收藏0
其他版本

除非特别说明，本系统中所有内容都受版权保护，并保留所有权利。