arrow
返回

Density-based weighting for imbalanced regression

delete2021-07-07
delete74
delete
OA
AI
M
Michael Steininger *
K
Konstantin Kobs
P
Padraig Davidson
A
Anna Krause
A
Andreas Hotho
DOI:10.1007/s10994-021-06023-5delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
In many real world settings, imbalanced data impedes model performance of learning algorithms, like neural networks, mostly for rare cases. This is especially problematic for tasks focusing on these rare occurrences. For example, when estimating precipitation, extreme rainfall events are scarce but important considering their potential consequences. While there are numerous well studied solutions for classification settings, most of them cannot be applied to regression easily. Of the few solutions for regression tasks, barely any have explored cost-sensitive learning which is known to have advantages compared to sampling-based methods in classification tasks. In this work, we propose a sample weighting approach for imbalanced regression datasets called DenseWeight and a cost-sensitive learning approach for neural network regression with imbalanced data called DenseLoss based on our weighting scheme. DenseWeight weights data points according to their target value rarities through kernel density estimation (KDE). DenseLoss adjusts each data point's influence on the loss according to DenseWeight, giving rare data points more influence on model training compared to common data points. We show on multiple differently distributed datasets that DenseLoss significantly improves model performance for rare data points through its density-based weighting scheme. Additionally, we compare DenseLoss to the state-of-the-art method SMOGN, finding that our method mostly yields better performance. Our approach provides more control over model training as it enables us to actively decide on the trade-off between focusing on common or rare cases through a single hyperparameter, allowing the training of better models for rare data points.
Keyword:
Imbalanced regression
Cost-sensitive learning
Sample weighting
Kernel-density estimation
Supervised learning
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Machine Learning 封面图
Machine Learning
IF:
2.9
论文数:
2.7K
被引数:
3.4W

机构

U
University of Wurzburg
学者数:
2.5W
论文数: 2.0W
被引数: 2.5W
引用论文

引用论文

Arsenic removal from aqueous solutions by adsorption using novel MIL-53(Fe) as a highly efficient adsorbent使用新型MIL-53(Fe) 作为高效吸附剂通过吸附从水溶液中去除砷
err2015-01-01
err0
PREAI
errTuan. A. Vu; Giang. H. Le; Canh. D. Dao; Lan. Q. Dang; Kien. T. Nguyen; Quang. K. Nguyen; Phuong. T. Dang; Hoa. T. K. Tran; Quang. T. Duong; Tuyen. V. Nguyen; Gun. D. Lee
err分享
err收藏
PECVD-grown carbon nanotubes on silicon substrates with a nickel-seeded tip-growth structure具有镍种子尖端生长结构的硅衬底上的PECVD生长的碳纳米管
err2006-07-01
err0
PREAI
errY. Abdi; J. Koohsorkhi; J. Derakhshandeh; S. Mohajerzadeh; H. Hoseinzadegan; M.D. Robertson; J.C. Bennett; X. Wu; H. Radamson
err分享
err收藏
An extended tuning method for cost-sensitive regression and forecasting
err2011-06-01
err36
PREAI
errZhao, Huimin; Sinha, Atish P.; Bansal, Gaurav
err分享
err收藏
学者 查看更多内容