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

UDRN:用于特征选择和特征投影的统一降维神经网络

UDRN: Unified Dimensional Reduction Neural Network for Feature Selection and Feature Projection

Zhejiang University(浙江大学);Westlake University(西湖大学);Westlake Institute for Advanced Study(西湖高等研究院);China Telecom Corporation Limited, Hangzhou Branch(中国电信股份有限公司杭州分公司)

Zelin Zang, Yongjie Xu, Linyan Lu, Yulan Geng, Senqiao Yang, Stan Z. Li

摘要:

降维技术是将高维数据映射到一个低维的潜在空间,并以最小化的优化目标为优化目标。DR方法通常分为特征选择FS和特征投影FP。FS侧重于选择维度的关键子集,但有破坏数据分布(结构)的风险。另一方面,FP将所有输入特征组合到低维空间中,旨在保持数据结构,但缺乏可解释性和稀疏性。金融服务和金融产品传统上是不相容的类别;因此,它们没有被统一成一个友好的框架。我们建议理想的DR方法将FS和FP结合到统一的端到端流形学习框架中,同时执行基本特征发现,同时维护潜在空间中数据样本之间的内在关系。在这项工作中,我们开发了一个统一的框架,统一降维神经网络~(UDRN),以兼容的,端到端的方式集成FS和FP。我们改进了神经网络的结构,分别实现FS和FP任务使用两个堆叠的子网络。此外,为了提高该方法在处理大规模特征数据集时的泛化能力,设计了DR过程的数据增强,并设计了与数据增强相配合的损失函数。在四个图像和四个生物数据集上的大量实验结果,包括非常高维的数据,证明了DRN相对于现有方法(FS,FP和FS\&FP管道)的优势,特别是在分类和可视化等下游任务中。

英文摘要:

Dimensional reduction~(DR) maps high-dimensional data into a lower dimensions latent space with minimized defined optimization objectives. The DR method usually falls into feature selection~(FS) and feature projection~(FP). FS focuses on selecting a critical subset of dimensions but risks destroying the data distribution (structure). On the other hand, FP combines all the input features into lower dimensions space, aiming to maintain the data structure; but lacks interpretability and sparsity. FS and FP are traditionally incompatible categories; thus, they have not been unified into an amicable framework. We propose that the ideal DR approach combines both FS and FP into a unified end-to-end manifold learning framework, simultaneously performing fundamental feature discovery while maintaining the intrinsic relationships between data samples in the latent space. In this work, we develop a unified framework, Unified Dimensional Reduction Neural-network~(UDRN), that integrates FS and FP in a compatible, end-to-end way. We improve the neural network structure by implementing FS and FP tasks separately using two stacked sub-networks. In addition, we designed data augmentation of the DR process to improve the generalization ability of the method when dealing with extensive feature datasets and designed loss functions that can cooperate with the data augmentation. Extensive experimental results on four image and four biological datasets, including very high-dimensional data, demonstrate the advantages of DRN over existing methods~(FS, FP, and FS\&FP pipeline), especially in downstream tasks such as classification and visualization.

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

↑