arXiv:2409.19396cs.LGcs.CV2024-09被引 4

用典型相关性约束训练深度网络,提升多视图表示学习效果

Canonical Correlation Guided Deep Neural Network

  • 以典型相关性为约束而非目标,更灵活适应工程任务
  • 在MNIST上重建误差低于DCCA和DCCAE,提升显著
  • 适用于故障诊断与寿命预测等工业场景

学习两组数据视图的表示,使所得表示具有高度线性相关性,是机器学习中的重要目标。本文提出一种典型相关性引导的学习框架(CCDNN),可由深度神经网络实现,用于学习此类相关表示。该方法是多元统计分析(MVA)与机器学习的创新融合,将传统MVA转化为端到端神经网络架构。与线性CCA、核CCA及深度CCA不同,本方法不以最大化相关性为目标,而是将典型相关性作为约束条件,保留相关表示学习能力的同时,更聚焦于优化形式所赋予的重建、分类与预测等工程任务。此外,为减少相关性带来的冗余,设计了冗余过滤器。实验表明,在MNIST数据集上,CCDNN的均方误差和平均绝对误差均优于DCCA与DCCAE。同时,该方法在工业故障诊断与剩余使用寿命预测任务中表现优异,显著超越现有方法。附录还展示了结合残差连接实现更深网络的扩展方案。

原文摘要 · Abstract (English)

Learning representations of two views of data such that the resulting representations are highly linearly correlated is appealing in machine learning. In this paper, we present a canonical correlation guided learning framework, which allows to be realized by deep neural networks (CCDNN), to learn such a correlated representation. It is also a novel merging of multivariate analysis (MVA) and machine learning, which can be viewed as transforming MVA into end-to-end architectures with the aid of neural networks. Unlike the linear canonical correlation analysis (CCA), kernel CCA and deep CCA, in the proposed method, the optimization formulation is not restricted to maximize correlation, instead we make canonical correlation as a constraint, which preserves the correlated representation learning ability and focuses more on the engineering tasks endowed by optimization formulation, such as reconstruction, classification and prediction. Furthermore, to reduce the redundancy induced by correlation, a redundancy filter is designed. We illustrate the performance of CCDNN on various tasks. In experiments on MNIST dataset, the results show that CCDNN has better reconstruction performance in terms of mean squared error and mean absolute error than DCCA and DCCAE. Also, we present the application of the proposed network to industrial fault diagnosis and remaining useful life cases for the classification and prediction tasks accordingly. The proposed method demonstrates superior performance in both tasks when compared to existing methods. Extension of CCDNN to much more deeper with the aid of residual connection is also presented in appendix.

多视图学习典型相关深度网络工业诊断

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。