Transfer Learning Under High-Dimensional Network Convolutional Regression Model

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wang, Liyuan, Chen, Jiachen, Lunetta, Kathryn L., Huang, Danyang, Cheng, Huimin, Mukherjee, Debarghya
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910921522151424
author Wang, Liyuan
Chen, Jiachen
Lunetta, Kathryn L.
Huang, Danyang
Cheng, Huimin
Mukherjee, Debarghya
author_facet Wang, Liyuan
Chen, Jiachen
Lunetta, Kathryn L.
Huang, Danyang
Cheng, Huimin
Mukherjee, Debarghya
contents Transfer learning enhances model performance by utilizing knowledge from related domains, particularly when labeled data is scarce. While existing research addresses transfer learning under various distribution shifts in independent settings, handling dependencies in networked data remains challenging. To address this challenge, we propose a high-dimensional transfer learning framework based on network convolutional regression (NCR), inspired by the success of graph convolutional networks (GCNs). The NCR model incorporates random network structure by allowing each node's response to depend on its features and the aggregated features of its neighbors, capturing local dependencies effectively. Our methodology includes a two-step transfer learning algorithm that addresses domain shift between source and target networks, along with a source detection mechanism to identify informative domains. Theoretically, we analyze the lasso estimator in the context of a random graph based on the Erdos-Renyi model assumption, demonstrating that transfer learning improves convergence rates when informative sources are present. Empirical evaluations, including simulations and a real-world application using Sina Weibo data, demonstrate substantial improvements in prediction accuracy, particularly when labeled data in the target domain is limited.
format Preprint
id arxiv_https___arxiv_org_abs_2504_19979
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Transfer Learning Under High-Dimensional Network Convolutional Regression Model
Wang, Liyuan
Chen, Jiachen
Lunetta, Kathryn L.
Huang, Danyang
Cheng, Huimin
Mukherjee, Debarghya
Machine Learning
Methodology
Transfer learning enhances model performance by utilizing knowledge from related domains, particularly when labeled data is scarce. While existing research addresses transfer learning under various distribution shifts in independent settings, handling dependencies in networked data remains challenging. To address this challenge, we propose a high-dimensional transfer learning framework based on network convolutional regression (NCR), inspired by the success of graph convolutional networks (GCNs). The NCR model incorporates random network structure by allowing each node's response to depend on its features and the aggregated features of its neighbors, capturing local dependencies effectively. Our methodology includes a two-step transfer learning algorithm that addresses domain shift between source and target networks, along with a source detection mechanism to identify informative domains. Theoretically, we analyze the lasso estimator in the context of a random graph based on the Erdos-Renyi model assumption, demonstrating that transfer learning improves convergence rates when informative sources are present. Empirical evaluations, including simulations and a real-world application using Sina Weibo data, demonstrate substantial improvements in prediction accuracy, particularly when labeled data in the target domain is limited.
title Transfer Learning Under High-Dimensional Network Convolutional Regression Model
topic Machine Learning
Methodology
url https://arxiv.org/abs/2504.19979