arXiv Machine Learning

The General Theory of Localization Methods

The paper introduces a general machine learning framework called the localization method, built on localization kernels and local means that underpin self‑attention. It formally defines the framework through a local‑ized model and a localization trick, and systematically connects it to a wide array of existing models such as kernel methods, MeanShift, Hopfield networks, LLE, fuzzy inference, and denoising autoencoders. The authors also extend the framework with adaptive kernels, hierarchical local models, and non‑local models, showing that the Transformer can be constructed using hierarchical local models, thereby unifying and generalizing state‑of‑the‑art architectures.

arXiv AI
Sep 30

AdaKerNet: Neural Kernel Decoding for Task-Adaptive Prediction with Multimodal Large Models

AdaKerNet is a task‑adaptive neural kernel decoder that operates on frozen multimodal representations from large foundation models, without requiring access to the models’ parameters. It learns Lipschitz‑controlled multimodal features, a reference kernel providing a soft structural prior, and a lightweight nonlinear predictor that deforms this structure. Experiments on four multimodal large language models and diverse input modalities show consistent improvements over baseline decoders, achieving up to 41% error reduction in scarce‑label settings.

By Konstantinos D. Polyzos, Eleni Oikonomou, Tara Javidi
arXiv Machine Learning
Sep 21

Diagonalized Attention for Individualized Regression: Latent-Row Localization and Prediction

The paper introduces an individualized sparse regression framework for matrix‑valued covariates, where each observation has its own relevant rows while regression effects are shared across the population. It proposes a diagonalized attention mechanism that uses query–key scores to localize sample‑specific signal rows and a value matrix for downstream regression, achieving a parameter dimension independent of sample size. The authors provide existence theorems guaranteeing recovery of latent rows under score‑separation and concentration conditions, and demonstrate strong prediction, localization, and classification performance in simulations and real sentiment analysis.

By Borui Peng, Liwei Lin, Feifei Wang, Long Feng
arXiv Machine Learning
Aug 27

Cubit: Token Mixer with Kernel Ridge Regression

The paper introduces Cubit, a Transformer‑style architecture that replaces the standard attention mechanism with Kernel Ridge Regression (KRR). By interpreting attention as Nadaraya‑Watson regression, Cubit incorporates the closed‑form KRR solution, combining kernel‑based value aggregation with normalization via the inverse kernel matrix. The authors also propose a Limited‑Range Rescale (LRR) to stabilize training and report that Cubit shows improved long‑sequence modeling, with gains increasing as training sequence length grows.

By Chuanyang Zheng, Jiankai Sun, Yihang Gao, Yuehao Wang, Liangchen Tan, Mac Schwager, Anderson Schneider, Yuriy Nevmyvaka, Xiaodong Liu
arXiv Machine Learning
Aug 26

Generalization, memorization, and overfitting for diffusion models trained in the lazy high-dimensional regime

The paper investigates diffusion models trained in a lazy high‑dimensional regime, extending benign overfitting theory to generative settings. By analyzing denoising score matching in a vector‑valued RKHS with an inner‑product kernel, the authors derive exact risk trajectories under gradient flow when the number of samples scales proportionally with dimensionality. These trajectories reveal three distinct phases—spectral generalization, noise‑dominated interpolation, and empirical Bayes memorization—whose interplay shapes the distribution of generated samples.

By Hugo Latourelle-Vigeant, Sinho Chewi, Aram-Alexandre Pooladian, John Sous, Theodor Misiakiewicz
arXiv Machine Learning
6d ago

Flow-based Spectral Kernel Learning for Nonstationary Attention

The paper introduces Flow-based Spectral Kernel Attention (FSKA), a framework that learns nonstationary attention kernels using spectral density learning and normalizing flows. FSKA replaces traditional query–key projections with a simplified design that retains fixed orthogonal value projections, leading to a more parameter‑efficient architecture. Experiments on classification tasks demonstrate competitive or improved performance while reducing trainable projection parameters by roughly 40%–47% and preserving linear‑attention memory scaling.

By Zi Yang, Ying Li, Alejandro Lancho, Dar\'{i}o C. Larese, Pablo M. Olmos, Michael Minyi Zhang
arXiv Machine Learning
Jun 2

Normalized Relevance Measure as a Unifying Framework to Explain Neural Network Latent Structures

arXiv:2606. 00557v1 Announce Type: new Abstract: To understand how a neural network (NN) functions and makes predictions, it has become increasingly clear that analyzing only the input domain is insufficient -- one must also examine its internal inference mechanisms to capture the complete picture.

By Ping Xiong, Thomas Schnake, Gr\'egoire Montavon, Klaus-Robert M\"uller, Shinichi Nakajima