activity
20192025
most citedFinding Patient Zero: Learning Contagion Source with Graph Neural Networks

28 citations · 48 across the 10 of their papers we have counts for

collaborators
Showing cs.LGShow all

12 papers · 1 filter

cs.LG2025

NRGPT: An Energy-based Alternative for GPT

Nima Dehmamy, Benjamin Hoover, Bishwajit Saha +3

Generative Pre-trained Transformer (GPT) architectures are the most popular design for language modeling. Energy-based modeling is a different paradigm that views inference as a dy…

cs.LG2025

Understanding Mode Connectivity via Parameter Space Symmetry

Bo Zhao, Nima Dehmamy, Robin Walters +1

Neural network minima are often connected by curves along which train and test loss remain nearly constant, a phenomenon known as mode connectivity. While this property has enabled…

cs.LG2025

Small Models, Smarter Learning: The Power of Joint Task Training

Csaba Both, Benjamin Hoover, Hendrik Strobelt +4

Multi-task learning improves generalization, but when does it reduce the model capacity required to learn? We provide a systematic study of how joint training affects the learning…

cs.LG2025

Discovering Symbolic Differential Equations with Symmetry Invariants

Jianke Yang, Manu Bhat, Bryan Hu +4

Discovering symbolic differential equations from data uncovers fundamental dynamical laws underlying complex systems. However, existing methods often struggle with the vast search…

cs.LG2025

AtlasD: Automatic Local Symmetry Discovery

Manu Bhat, Jonghyun Park, Jianke Yang +3

Existing symmetry discovery methods predominantly focus on global transformations across the entire system or space, but they fail to consider the symmetries in local neighborhoods…

cs.LG20241 cited

Symmetry-Informed Governing Equation Discovery

Jianke Yang, Wang Rao, Nima Dehmamy +2

Despite the advancements in learning governing differential equations from observations of dynamical systems, data-driven methods are often unaware of fundamental physical laws, su…