arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2104.06313cs.IRcs.AIcs.LG

SetConv: A New Approach for Learning from Imbalanced Data

The University of Texas at Dallas(德克萨斯大学达拉斯分校);IBM T.J. Watson(IBM T.J. Watson研究中心)

Yang Gao, Yi-Fan Li, Yu Lin, Charu Aggarwal, Latifur Khan

英文摘要:

For many real-world classification problems, e.g., sentiment classification, most existing machine learning methods are biased towards the majority class when the Imbalance Ratio (IR) is high. To address this problem, we propose a set convolution (SetConv) operation and an episodic training strategy to extract a single representative for each class, so that classifiers can later be trained on a balanced class distribution. We prove that our proposed algorithm is permutation-invariant despite the order of inputs, and experiments on multiple large-scale benchmark text datasets show the superiority of our proposed framework when compared to other SOTA methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑