Nandan Kumar Jha
Nandan Kumar Jha
Home
Research
Publications
Highlights
Talks
Media
Contact
3
Same Architecture, Different Capacity: Optimizer-Induced Spectral Scaling Laws
Shows that optimizers can determine how much nominal FFN width becomes realized spectral capacity, even when validation loss is matched.
Nandan Kumar Jha
,
Brandon Reagen
PDF
Cite
Code
Project
Blog
A Random Matrix Theory Perspective on the Learning Dynamics of Multi-head Latent Attention
Uses random-matrix tools to analyze how multi-head latent attention evolves during training, revealing capacity bottlenecks and representation-geometry shifts.
Nandan Kumar Jha
,
Brandon Reagen
PDF
Cite
News
AERO: Entropy-Guided Attention for Private LLM Inference
Develops entropy-guided attention and hierarchical entropy regularization for efficient private LLM inference with reduced nonlinearities.
Nandan Kumar Jha
,
Brandon Reagen
Cite
Code
Poster
Video
Earlier arXiv
Press release
Regularizing the Entropy Landscape of Self-Attention: Towards a Soft Inductive Bias in LLMs
Studies entropy regularization for self-attention as a soft inductive bias in large language models.
Nandan Kumar Jha
,
Brandon Reagen
Cite
Workshop
OpenReview
Sisyphus: A Cautionary Tale of Using Low-Degree Polynomial Activations in Privacy-Preserving Deep Learning
Privacy concerns in client-server machine learning have given rise to private inference (PI), where neural inference occurs directly on …
Karthik Garimella
,
Nandan Kumar Jha
,
Brandon Reagen
PDF
Cite
Code
Poster
PPML Proceeding
CryptoNite: Revealing the Pitfalls of End-to-End Private Inference at Scale
In this paper we demonstrated that how the current trend in private inference myopically optimized the performance only for zero arrival rate; in particular, they have developed the mechanism to mitigate the bottlenecked caused by non-linearity in neural networks. However, in a real-world scenario when inference request comes even with a moderate arrival rate the homomorphic encryption becomes the main bottleneck since we can no longer pre-process it in the offline computation phase.
Karthik Garimella
,
Nandan Kumar Jha
,
Zahra Ghodsi
,
Siddharth Garg
,
Brandon Reagen
PDF
Cite
ReLU's Revival: On the Entropic Overload in Normalization-Free Large Language Models
LayerNorm is a critical component in modern large language models (LLMs) for stabilizing training and ensuring smooth optimization. …
Nandan Kumar Jha
,
Brandon Reagen
PDF
Cite
Code
Poster
Workshop
Entropy-Guided Attention for Private LLMs
We introduce an information-theoretic framework to characterize the role of nonlinearities in decoder-only language models, laying a …
Nandan Kumar Jha
,
Brandon Reagen
PDF
Cite
Code
Poster
Long video
Press release
Cite
×