arXiv · 2602.17423
Convergence Analysis of Two-Layer Neural Networks under Gaussian Input Masking
Abstract
We investigate the convergence guarantee of two-layer neural network training with Gaussian randomly masked inputs. This scenario corresponds to Gaussian dropout at the input level, or noisy input training common in sensor networks, privacy-preserving training, and federated learning, where each user may have access to partial or corrupted features. Using a Neural Tangent Kernel (NTK) analysis, we demonstrate that training a two-layer ReLU network with Gaussian randomly masked inputs achieves linear convergence up to an error region proportional to the mask's variance. A key technical contribution is resolving the randomness within the non-linear activation, a problem of independent interest.
Explore related subjects
Keep this discovery
Afroditi Kolomvaki, Fangshuo Liao, Evan Dramko, Ziyun Guang, Anastasios Kyrillidis. 2026-02-19. Convergence Analysis of Two-Layer Neural Networks under Gaussian Input Masking. https://arxiv.org/abs/2602.17423
Cite the original work for its findings. Save a collection to share your selection of sources.