arrow
Return

Penalized multiple distribution selection method for imbalanced data classification

delete2020-05-01
delete12
PRE
AI
石戈 (Ge Shi)
C
Chong Feng *
徐文福 (Wenfu Xu)
L
Lejian Liao
黄河燕 cover
黄河燕 (Heyan Huang)
DOI:10.1016/j.knosys.2020.105833delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In reality, the amount of data from different categories varies significantly, which results in learning bias towards prominent classes, hindering the overall classification performance. In this paper, by proving that traditional classification methods that use single softmax distribution are limited for modeling complex and imbalanced data, we propose a general Multiple Distribution Selection (MDS) method for imbalanced data classification. MDS employs a mixture distribution that is composed of a single softmax distribution and a set of degenerate distributions to model imbalanced data. Furthermore, a dynamic distribution selection method, based on L-1 regularization, is also proposed to automatically determine the weights of distributions. In addition, the corresponding two-stage optimization algorithm is designed to estimate the parameters of models. Extensive experiments conducted on three widely used benchmark datasets (IMDB, ACE2005, 20NewsGroups) show that our proposed mixture method outperforms previous methods. Moreover, under highly imbalanced setting, our method achieves up to a 4.1 absolute F1 gain over high-performing baselines. (C) 2020 Elsevier B.V. All rights reserved.
Keywords:
Imbalance training
Knowledge extraction
Mixture distribution
Distribution selection
Text classification
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

K
Knowledge-Based Systems
IF:
7.6
Papers:
1.2W
Citations:
4.5W

Organization

C
China Jiliang University
Scholars:
9.8K
Papers: 6.3K
Citations: 7.2K
B
beijing institute of technology
Scholars:
5.4W
Papers: 3.9W
Citations: 63