返回
Borderline-margin loss based deep metric learning framework for imbalanced data
DOI:10.1007/s10489-022-03494-4.png)
摘要
En 中文
The imbalanced data suffer the problem where minority class is under-represented compared with majority ones. Traditional imbalanced learning algorithms only consider the class imbalance while ignoring the class overlap, which leads to an undesirable accuracy for minority samples in overlapping regions. Considering the above issue, we propose a deep metric framework with borderline-margin loss (DMFBML) for improving the intra-class coherence and inter-class difference in overlapping regions. Firstly, a flexible borderline margin is designed for each minority sample, which is adaptively adjusted according to the neighborhood's label. The proposed margin enables to discriminate minority samples with varying overlap degrees, which significantly preserves the valuable information of classification boundary. The input data is then reconstructed into training triplets set to generate more metric constraints for minority samples, thereby increasing the difference in overlapping regions. Finally, a neural network with DMFBML is presented to achieve a better classifier performance on imbalanced data. The proposed method is verified by the comparative experiments on six synthetic datasets and eleven actual datasets.
Keyword:
Imbalanced classification
Class imbalance
Class overlap
Deep metric framework
Borderline-margin loss
期刊
IF:
3.5
论文数:
7.6K
被引数:
1.7W
机构
引用论文
Global and local semantics-preserving based deep hashing for cross-modal retrieval基于全局和局部语义保持的深度哈希跨模态检索
NEUROCOMPUTING
IF6.5
Cost sensitive active learning using bidirectional gated recurrent neural networks for imbalanced fault diagnosis
NEUROCOMPUTING
IF6.5

