Saved in:
Bibliographic Details
Main Authors: Li, Yongxin, Wang, Yifan, Lin, Zhongshuo, Xie, Hehu
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2406.09694
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909313589575680
author Li, Yongxin
Wang, Yifan
Lin, Zhongshuo
Xie, Hehu
author_facet Li, Yongxin
Wang, Yifan
Lin, Zhongshuo
Xie, Hehu
contents This paper introduces a tensor neural network (TNN) to address nonparametric regression problems, leveraging its distinct sub-network structure to effectively facilitate variable separation and enhance the approximation of complex, high-dimensional functions. The TNN demonstrates superior performance compared to conventional Feed-Forward Networks (FFN) and Radial Basis Function Networks (RBN) in terms of both approximation accuracy and generalization capacity, even with a comparable number of parameters. A significant innovation in our approach is the integration of statistical regression and numerical integration within the TNN framework. This allows for efficient computation of high-dimensional integrals associated with the regression function and provides detailed insights into the underlying data structure. Furthermore, we employ gradient and Laplacian analysis on the regression outputs to identify key dimensions influencing the predictions, thereby guiding the design of subsequent experiments. These advancements make TNN a powerful tool for applications requiring precise high-dimensional data analysis and predictive modeling.
format Preprint
id arxiv_https___arxiv_org_abs_2406_09694
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle An Efficient Approach to Regression Problems with Tensor Neural Networks
Li, Yongxin
Wang, Yifan
Lin, Zhongshuo
Xie, Hehu
Machine Learning
62J02, 68T05
This paper introduces a tensor neural network (TNN) to address nonparametric regression problems, leveraging its distinct sub-network structure to effectively facilitate variable separation and enhance the approximation of complex, high-dimensional functions. The TNN demonstrates superior performance compared to conventional Feed-Forward Networks (FFN) and Radial Basis Function Networks (RBN) in terms of both approximation accuracy and generalization capacity, even with a comparable number of parameters. A significant innovation in our approach is the integration of statistical regression and numerical integration within the TNN framework. This allows for efficient computation of high-dimensional integrals associated with the regression function and provides detailed insights into the underlying data structure. Furthermore, we employ gradient and Laplacian analysis on the regression outputs to identify key dimensions influencing the predictions, thereby guiding the design of subsequent experiments. These advancements make TNN a powerful tool for applications requiring precise high-dimensional data analysis and predictive modeling.
title An Efficient Approach to Regression Problems with Tensor Neural Networks
topic Machine Learning
62J02, 68T05
url https://arxiv.org/abs/2406.09694