← Search

Nima Dehmamy

14 accepted papers

2026

NRGPT: An Energy-based Alternative for GPT

ICLR 2026poster

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 dynamical process operating on an energy landscape. We propose a minimal modification of the GPT setting to unify it with…

Cited by 0SourceScholar
2025

AtlasD: Automatic Local Symmetry Discovery

ICML 2025poster

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. This may result in the reported symmetry group being a misrepresentation of the true symmetry. In this paper, we forma…

2025

Large Language Models Are Natural Video Popularity Predictors

ACL 2025finding

Predicting video popularity is often framed as a supervised learning task, relying heavily on meta-information and aggregated engagement data. However, video popularity is shaped by complex cultural and social factors that such approaches often overlook. We argue that Large Language Models (LLMs), w…

2025

Symmetry-Driven Discovery of Dynamical Variables in Molecular Simulations

ICML 2025poster

We introduce a novel approach for discovering effective degrees of freedom (DOF) in molecular dynamics simulations by mapping the DOF to approximate symmetries of the energy landscape. Unlike most existing methods, we do not require trajectory data but instead rely on knowledge of the forcefield (en…

Cited by 0SourcePDFScholar
2024

Neural Network Reparametrization for Accelerated Optimization in Molecular Simulations

NeurIPS 2024poster

We propose a novel approach to molecular simulations using neural network reparametrization, which offers a flexible alternative to traditional coarse-graining methods. Unlike conventional techniques that strictly reduce degrees of freedom, the complexity of the system can be adjusted in our model,…

Cited by 0SourcePDFScholar
2024

Symmetry-Informed Governing Equation Discovery

NeurIPS 2024poster

Despite the advancements in learning governing differential equations from observations of dynamical systems, data-driven methods are often unaware of fundamental physical laws, such as frame invariance. As a result, these algorithms may search an unnecessarily large space and discover less accurate…

2023

Symmetries, Flat Minima, and the Conserved Quantities of Gradient Flow

ICLR 2023poster

Empirical studies of the loss landscape of deep networks have revealed that many local minima are connected through low-loss valleys. Yet, little is known about the theoretical origin of such valleys. We present a general framework for finding continuous symmetries in the parameter space, which carv…

2021

Automatic Symmetry Discovery with Lie Algebra Convolutional Network

NeurIPS 2021poster

Existing equivariant neural networks require prior knowledge of the symmetry group and discretization for continuous groups. We propose to work with Lie algebras (infinitesimal generators) instead of Lie groups. Our model, the Lie algebra convolutional network (L-conv) can automatically discover sym…

2019

Direct Estimation of Weights and Efficient Training of Deep Neural Networks without SGD

ICASSP 2019accepted

We argue that learning a hierarchy of features in a hierarchical dataset requires lower layers to approach convergence faster than layers above them. We show that, if this assumption holds, we can analytically approximate the outcome of stochastic gradient descent (SGD) for each layer. We find that…

Cited by 0SourceScholar
2019

Understanding the Representation Power of Graph Neural Networks in Learning Graph Topology

NeurIPS 2019poster

To deepen our understanding of graph neural networks, we investigate the representation power of Graph Convolutional Networks (GCN) through the looking glass of graph moments, a key property of graph topology encoding path of various lengths. We find that GCNs are rather restrictive in learning gra…