← Search

Tanya Marwah

11 accepted papers

2026

Probabilistic Retrofitting of Learned Simulators

ICML 2026poster

Dominant approaches for modelling Partial Differential Equations (PDEs) rely on deterministic predictions, yet many physical systems of interest are inherently chaotic and uncertain. While training probabilistic models from scratch is possible, it is computationally expensive and fails to leverage t…

Cited by 0SourceScholar
2026

Protein Design with Agent Rosetta: A Case Study for Specialized Scientific Agents

ICML 2026poster

Large language models (LLMs) are capable of emulating reasoning and using tools, creating opportunities for autonomous agents that execute complex scientific tasks. Protein design provides a natural testbed: although machine learning (ML) methods achieve strong results, these are largely restricted …

Cited by 0SourceScholar
2026

Walrus: A Cross-domain Foundation Model for Continuum Dynamics

ICML 2026spotlight

Foundation models have transformed machine learning for language and vision, but achieving comparable impact in physical simulation remains a challenge. Data heterogeneity and unstable long-term dynamics inhibit learning from sufficiently diverse dynamics, while varying resolutions and dimensionalit…

Cited by 0SourceScholar
2025

On the Benefits of Memory for Modeling Time-Dependent PDEs

ICLR 2025oral

Data-driven techniques have emerged as a promising alternative to traditional numerical methods for solving PDEs. For time-dependent PDEs, many approaches are Markovian---the evolution of the trained system only depends on the current state, and not the past states. In this work, we investigate the…

Cited by 3SourcePDFScholar
2025

Predicting partially observable dynamical systems via diffusion models with a multiscale inference scheme

NeurIPS 2025poster

Conditional diffusion models provide a natural framework for probabilistic prediction of dynamical systems and have been successfully applied to fluid dynamics and weather prediction. However, in many settings, the available information at a given time represents only a small fraction of what is nee…

Cited by 0SourceScholar
2025

Towards characterizing the value of edge embeddings in Graph Neural Networks

ICML 2025poster

Graph neural networks (GNNs) are the dominant approach to solving machine learning problems defined over graphs. Despite much theoretical and empirical work in recent years, our understanding of finer-grained aspects of architectural design for GNNs remains impoverished. In this paper, we consider t…

Cited by 1SourcePDFScholar
2023

Deep Equilibrium Based Neural Operators for Steady-State PDEs

NeurIPS 2023poster

Data-driven machine learning approaches are being increasingly used to solve partial differential equations (PDEs). They have shown particularly striking successes when training an operator, which takes as input a PDE in some family, and outputs its solution. However, the architectural design space,…

Cited by 7SourcePDFScholar
2023

Disentangling the Mechanisms Behind Implicit Regularization in SGD

ICLR 2023poster

A number of competing hypotheses have been proposed to explain why small-batch Stochastic Gradient Descent (SGD) leads to improved generalization over the full-batch regime, with recent work crediting the implicit regularization of various quantities throughout training. However, to date, empirical…

2023

Neural Network Approximations of PDEs Beyond Linearity: A Representational Perspective

ICML 2023poster

A burgeoning line of research has developed deep neural networks capable of approximating the solutions to high dimensional PDEs, opening related lines of theoretical inquiry focused on explaining how it is that these models appear to evade the curse of dimensionality. However, most theoretical anal…

Cited by 12SourcePDFScholar
2021

Parametric Complexity Bounds for Approximating PDEs with Neural Networks

NeurIPS 2021spotlight

Recent experiments have shown that deep networks can approximate solutions to high-dimensional PDEs, seemingly escaping the curse of dimensionality. However, questions regarding the theoretical basis for such approximations, including the required network size remain open. In this paper, we investig…

Cited by 26SourcePDFScholar