arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.09596cs.LG

Robustness to Missing Features using Hierarchical Clustering with Split Neural Networks

Massachusetts Institute of Technology(麻省理工学院);BITS Pilani Goa Campus(BITS Pilani果阿校区)

Rishab Khincha, Utkarsh Sarawgi, Wazeer Zulfikar, Pattie Maes

英文摘要:

The problem of missing data has been persistent for a long time and poses a major obstacle in machine learning and statistical data analysis. Past works in this field have tried using various data imputation techniques to fill in the missing data, or training neural networks (NNs) with the missing data. In this work, we propose a simple yet effective approach that clusters similar input features together using hierarchical clustering and then trains proportionately split neural networks with a joint loss. We evaluate this approach on a series of benchmark datasets and show promising improvements even with simple imputation techniques. We attribute this to learning through clusters of similar features in our model architecture. The source code is available at https://github.com/usarawgi911/Robustness-to-Missing-Features

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

↑