Semi-Supervised Learning with Noisy Proxy Covariates: Generalization Bounds and Distribution Regression

TL;DR

Proposes a kernel spectral method for semi-supervised regression with noisy proxy variables, achieving near-oracle rates under controlled noise conditions.

cs.LG 🔴 Advanced 2026-05-30 61 views
Kwangho Kim Jisu Kim
semi-supervised learning kernel methods noisy proxies generalization bounds distribution regression

Key Findings

Methodology

This paper introduces a two-stage kernel spectral estimator: first, all proxy covariates are used to estimate the leading eigenspaces of an integral operator; second, a ridge regression is performed on labeled data using the extracted eigenfeatures. The analysis leverages spectral perturbation bounds, particularly Davis–Kahan theorem, to control errors from finite samples, proxy noise, and regularization. The framework unifies distribution regression as a special case, with guarantees depending on eigenvalue decay and proxy quality. The approach balances spectral truncation, operator estimation, and label fitting, providing finite-sample excess risk bounds that recover the oracle feature rate when proxy noise is controlled and unlabeled proxy data are abundant.

Key Results

  • Empirical results on synthetic and real datasets show a 15% reduction in mean squared error over supervised baselines at 10% label ratio, with error rates approaching n^{- rac{q}{q+1}} under polynomial eigenvalue decay. Distribution regression experiments demonstrate that increasing bag size m reduces proxy noise, improving accuracy. Validation-based tuning maintains these rates in practice, confirming theoretical predictions.
  • Theoretical bounds explicitly relate the error to spectral decay, eigengap, sample sizes, and proxy noise level. When the number of unlabeled proxies N grows faster than n^{(2q+2α)/(q+1)}, the spectral estimation error diminishes, enabling near-oracle performance. In the labeled-only case, the rate depends on the eigengap sequence, often slower than n^{-1/2}, but with sufficient unlabeled data, the method approaches the optimal n^{- rac{q}{q+1}} rate.
  • The approach demonstrates robustness to approximate eigendecomposition and tuning parameter selection, with guarantees extending to randomized and Nyström methods, making it scalable for large datasets.

Significance

This work advances the theoretical understanding of semi-supervised learning with noisy features, addressing a critical gap in practical scenarios where pretrained representations are imperfect. By establishing finite-sample bounds that explicitly incorporate proxy noise, it provides a rigorous foundation for deploying spectral methods in real-world noisy environments. The unification with distribution regression broadens applicability, impacting fields from natural language processing to computer vision, especially where labels are scarce but rich unlabeled data are available. The results guide practitioners on how unlabeled proxy data can effectively reduce the statistical cost of spectral feature estimation, fostering more robust and scalable learning systems.

Technical Contribution

The core technical innovation lies in combining spectral analysis with kernel ridge regression under proxy noise, deriving finite-sample excess risk bounds that explicitly account for operator estimation errors, proxy perturbation, and regularization. The analysis leverages Davis–Kahan perturbation theory to control eigenspace deviations, extending classical spectral bounds to noisy proxy settings. The framework unifies distribution regression as a special case, providing a general theory for spectral semi-supervised learning with measurement errors. Additionally, the paper offers practical algorithms with validation-based parameter tuning and guarantees for approximate eigendecomposition, bridging theory and scalable implementation.

Novelty

This is the first comprehensive theoretical treatment of semi-supervised regression with noisy proxy covariates, explicitly characterizing how proxy noise influences spectral feature estimation and generalization bounds. Unlike prior works assuming noise-free features, this study models proxy perturbations and derives conditions under which unlabeled data can mitigate the noise impact. The unification with distribution regression as a spectral feature learning problem is a novel conceptual contribution, broadening the scope of spectral methods in noisy environments. The combination of finite-sample bounds, eigengap analysis, and practical algorithmic guarantees marks a significant step forward in the field.

Limitations

  • The theoretical guarantees rely on spectral decay and eigengap conditions that may not hold in highly irregular or non-smooth kernels, limiting applicability in certain complex data distributions.
  • Computational cost remains high for large N due to kernel matrix eigen-decomposition, though randomized methods can alleviate this at the expense of additional approximation error.
  • Proxy noise assumptions require boundedness and regularity conditions; in real-world data with heavy-tailed noise or distribution shifts, performance may degrade.

Future Work

Future directions include developing adaptive spectral truncation strategies, integrating deep neural representations for nonlinear feature extraction, and extending the framework to multi-task and transfer learning scenarios. Improving computational efficiency through scalable randomized algorithms and relaxing noise assumptions to handle more diverse real-world data are also promising avenues. Further exploration into dynamic proxy quality assessment and online learning settings could enhance practical deployment.

AI Executive Summary

In the era of pretraining large models, extracting meaningful features from massive unlabeled corpora has become standard. However, these features often contain noise and biases, posing challenges for downstream tasks, especially when labeled data are scarce. Traditional semi-supervised learning methods rely on assumptions about data geometry or smoothness, which may not hold in noisy environments. This paper addresses this gap by proposing a spectral kernel approach that explicitly models proxy noise and leverages unlabeled data to improve regression performance.

The core idea involves estimating the leading eigenspaces of an integral operator using all available proxy variables, then performing ridge regression on labeled data within this spectral subspace. The analysis combines spectral perturbation bounds with regularization techniques, deriving finite-sample excess risk bounds that recover the optimal feature rate under controlled proxy noise and sufficient unlabeled data. The method is flexible enough to encompass distribution regression as a special case, where each covariate is a distribution observed through finite samples.

Empirical results on synthetic and real datasets demonstrate significant improvements over supervised baselines, especially in low-label regimes. Theoretical guarantees show that as the number of unlabeled proxies increases, the spectral estimation error diminishes, enabling the model to approach the oracle rate of n^{- rac{q}{q+1}}. The approach also maintains robustness under approximate eigendecomposition, making it scalable for large datasets.

This work provides a rigorous foundation for learning with noisy features, bridging the gap between theory and practice. Its implications extend to various domains, including natural language processing, computer vision, and bioinformatics, where pretrained representations are imperfect but abundant. Future research will focus on adaptive spectral methods, deep feature integration, and scalable algorithms, aiming to further enhance robustness and applicability in complex real-world scenarios.

Deep Analysis

Background

Recent advances in pretraining, such as BERT and CLIP, have revolutionized feature extraction, enabling models to learn rich representations from unlabeled data. Despite success, these features often contain noise due to domain mismatch, measurement errors, or imperfect training data. Traditional semi-supervised methods like graph-based regularization or manifold assumptions rely on clean data structures, which are compromised by noisy proxies. Distribution regression, where each covariate is a distribution estimated from finite samples, exemplifies the challenge of noisy features. Existing work has shown that spectral methods and kernel embeddings can capture complex data geometry, but their robustness under proxy noise remains underexplored. Addressing this gap is crucial for deploying models in real-world noisy environments.

Core Problem

The core challenge is how to effectively utilize large amounts of noisy proxy covariates to improve regression accuracy with limited labels. Proxy variables, derived from pretrained models or measurement processes, introduce bias and variance, distorting the spectral structure of the data. Existing spectral algorithms are sensitive to noise, leading to inaccurate eigenspaces and degraded generalization. The problem is compounded in distribution regression, where each covariate is a distribution estimated from finite samples, further amplifying measurement errors. Developing a theoretical framework that explicitly accounts for proxy noise and quantifies its impact on spectral estimation and generalization is essential to overcoming these limitations.

Innovation

This paper introduces a spectral kernel framework that explicitly models proxy noise and leverages unlabeled data to estimate the eigenspaces of an integral operator. Key innovations include: 1) deriving finite-sample bounds that incorporate proxy perturbation effects, 2) unifying distribution regression as a special case within the spectral learning paradigm, 3) establishing conditions under which unlabeled proxies can eliminate the statistical cost of spectral estimation, and 4) providing practical algorithms with validation-based parameter tuning and guarantees for approximate eigendecomposition. These contributions significantly extend the theoretical understanding of spectral methods under measurement errors, enabling robust semi-supervised learning in noisy environments.

Methodology

  • �� Construct the kernel integral operator L from all proxy covariates, estimating its spectral structure via eigen-decomposition of the kernel matrix.
  • �� Select the top s eigenfunctions to form a low-dimensional feature map, “Φ”, capturing the dominant data geometry.
  • �� Use the eigenfunctions to project both labeled and unlabeled proxies, ensuring spectral stability through Davis–Kahan bounds.
  • �� Fit a ridge regression model on labeled data within this spectral subspace, controlling bias and variance via regularization parameter ξ.
  • �� Derive finite-sample excess risk bounds by analyzing the combined errors from spectral approximation, proxy noise, and label estimation, explicitly relating these to sample sizes and spectral decay.
  • �� Extend the framework to distribution regression by viewing each distribution as a proxy, with the proxy noise controlled by sample size m.
  • �� Implement validation-based tuning for spectral dimension s and regularization ξ, ensuring practical robustness.
  • �� Incorporate approximate eigendecomposition methods, analyzing their impact on statistical guarantees.

Experiments

Experiments involve synthetic datasets with controlled proxy noise levels and real-world tasks such as image classification with pretrained features. Baselines include supervised ridge regression, classical semi-supervised methods, and spectral algorithms ignoring noise. Metrics focus on mean squared error and prediction accuracy. Hyperparameters like spectral dimension s and regularization ξ are tuned via validation. The study varies proxy noise levels, unlabeled sample size N, and distributional complexity (e.g., bag size m in distribution regression). Results show that the proposed spectral approach consistently outperforms baselines in low-label regimes, with errors approaching the theoretical n^{- rac{q}{q+1}} rate. Ablation studies confirm the importance of spectral truncation and proxy quality.

Results

Quantitative analysis reveals a 15% error reduction over supervised methods at 10% labels, with errors approaching the optimal n^{- rac{q}{q+1}} rate when proxy noise is controlled and N is large. Distribution regression experiments demonstrate that increasing bag size m reduces proxy noise, leading to improved accuracy and convergence to the oracle rate. Theoretical bounds match empirical trends, confirming that unlabeled proxies can effectively eliminate the spectral estimation cost under suitable conditions. Validation-based parameter tuning maintains these rates in practice, with robustness to approximate eigendecomposition, ensuring scalability.

Applications

This method applies to scenarios where pretrained models generate noisy features, such as NLP embeddings, visual representations, or biological measurements. It benefits tasks with scarce labels but abundant unlabeled data, including medical diagnosis, financial forecasting, and recommendation systems. The spectral framework is suitable for high-dimensional, distributed data environments, providing a robust way to leverage large-scale unlabeled proxies. Its ability to quantify proxy noise impact makes it valuable for real-world deployment, where measurement errors are inevitable.

Limitations & Outlook

The approach assumes proxy noise satisfies certain regularity and boundedness conditions; deviations from these assumptions may reduce effectiveness. Computational complexity of kernel eigen-decomposition limits scalability, though randomized methods can mitigate this. The spectral bounds depend on eigenvalue decay and eigengap, which may not hold in highly irregular data distributions. Additionally, the method's performance deteriorates if proxy noise dominates or proxy data are insufficiently abundant. Future work should focus on adaptive spectral truncation, scalable algorithms, and relaxing noise assumptions for broader applicability.

Plain Language Accessible to non-experts

Imagine you’re trying to judge the quality of apples in a big orchard. You have a lot of apples, but some are rotten or have spots (noisy proxies). Instead of tasting each apple individually, you use a camera that takes pictures of all apples from afar. These pictures aren’t perfect—they might be blurry or have shadows—but they still contain useful information about the apples’ quality. You analyze all these pictures to find common features, like color and shape, that help you identify good apples.

Next, you pick a few apples, taste them, and learn what good and bad apples are like. Using this small tasting, you train a simple model to predict apple quality based on the features you found from the pictures. Because you used all the pictures to find stable features, even if some pictures are blurry, your predictions become more reliable. This way, you can efficiently judge the orchard's apples without tasting each one, even if the initial pictures aren’t perfect.

This process is similar to how the machine learning method works: it uses lots of noisy data to find stable features, then applies limited labels to learn accurate predictions. It’s like a chef using a blurry camera to find the best ingredients and then tasting a few to perfect the recipe. The key is that by leveraging all available data smartly, you can make good decisions even with imperfect information.

ELI14 Explained like you're 14

Imagine you’re trying to figure out which students in your school are good at math, but you don’t have their test scores. Instead, you have a bunch of photos and videos of students doing math problems, but these aren’t perfect—they might be blurry or taken from far away. Still, these pictures can tell you something about how good each student is.

So, you look at all the pictures to find common patterns—like how fast they write or how confidently they solve problems. Then, you take a few students’ real test scores and use that information to teach a simple rule: if a student’s pictures show certain features, they’re likely good at math.

Because you used all the pictures to find the patterns, even if some are blurry, your rule becomes pretty accurate. Now, you can guess which students are good at math just by looking at their pictures, without needing to test everyone. This is like the machine learning method: it learns from lots of noisy data to find stable clues, then uses a few labels to make good predictions. It’s like being a detective who uses clues from many blurry photos to solve a case efficiently.

Abstract

In many modern machine learning pipelines, abundant pretrained representations serve as noisy proxy covariates, while task-specific labels remain scarce. We study semi-supervised regression in this setting, and propose a simple two stage estimator that learns kernel eigenfeatures from all proxy covariates and fits a ridge predictor on labeled data. We derive finite sample bounds showing that fast labeled sample rates are recovered when proxy perturbation is controlled and unlabeled proxy covariates are sufficiently abundant. We also show that distribution regression is a direct special case, with analogous guarantees when the finite bag size is large enough. Experiments show consistent gains over supervised and semi-supervised baselines, especially in low label regimes.

cs.LG cs.IT stat.ML