English

Diffusion-Guided Feature Selection via Nishimori Temperature: Noise-Based Spectral Embedding

Machine Learning 2026-04-28 v1

Abstract

We propose Noise-Based Spectral Embedding (NBSE), a physics-informed framework for selecting informative features from high-dimensional data without greedy search. NBSE constructs a sparse similarity graph on the samples and identifies the Nishimori temperature βN\beta_N the critical inverse temperature at which the Bethe Hessian becomes singular. The corresponding smallest eigenvector captures the dominant mode of an intrinsically degree-corrected diffusion process, naturally reweighting nodes to prevent hub dominance. By transposing the data matrix and applying NBSE in feature space, we obtain a one-dimensional spectral embedding that reveals groups of redundant or semantically related dimensions; balanced binning then selects one representative per group. We prove that coloured Gaussian perturbations shift βN\beta_N by at most O(σˉ2)O(\bar\sigma^2), guaranteeing robustness to measurement noise. Experiments on ImageNet embeddings from MobileNetV2 and EfficientNet-B4 show that NBSE preserves classification accuracy even under aggressive compression: on EfficientNet-B4 the accuracy drop is below 1%1\% when retaining only 30%30\% of features, outperforming ANOVA FF-test and random selection by up to 6.8%6.8\%.

Keywords

Cite

@article{arxiv.2604.24692,
  title  = {Diffusion-Guided Feature Selection via Nishimori Temperature: Noise-Based Spectral Embedding},
  author = {Vasiliy S. Usatyuk and Denis A. Sapozhnikov and Sergey I. Egorov},
  journal= {arXiv preprint arXiv:2604.24692},
  year   = {2026}
}

Comments

8 pages, 3 figures, extended version (with noise shift proof) of DSPA2026 article

R2 v1 2026-07-01T12:37:35.873Z