activity
20092026
most citedA Koopman framework for rare event simulation in stochastic differential equations

18 citations · 68 across the 27 of their papers we have counts for

collaborators

32 papers

cs.LG2026

Certifying Residual Architectures from Their Primitives: A Sharp Stability Threshold

Hyemin Gu, Michael Tyrrell, Tuhin Sahai +1

Whether a deep residual architecture trains stably is usually determined by training it, which is expensive and answers the question only for the architecture, depth, and floating-…

stat.ML2026

ISOMORPH: A Supply Chain Digital Twin for Simulation, Dataset Generation, and Forecasting Benchmarks

Zhizhen Zhang, Hyemin Gu, Benjamin J. Zhang +6

Open time-series forecasting (TSF) benchmarks cover retail, energy, weather, and traffic, but supply-chain logistics remains underserved. We introduce ISOMORPH, the first public di…

cs.LG2026

Vocabulary-size-independent Convergence of Discrete Diffusion Models: adjoint equations induce the right space

Kelvin Kan, Xingjian Li, Benjamin J. Zhang +3

Discrete diffusion has become a leading framework for generative modeling in various applications including language, vision, and biology. Existing convergence theory, however, exh…

cs.LG2026

Optimized Piecewise Affine Abstractions of Neural Networks with Learnable Activation Functions

Noah Schwartz, Chandra Kanth Nagesh, Sriram Sankaranarayanan +3

We present a generalized framework for the range verification of neural networks featuring non-linear activation functions. Our approach first constructs an ``optimized piecewise a…

cs.LG2025

Stability of Transformers under Layer Normalization

Kelvin Kan, Xingjian Li, Benjamin J. Zhang +4

Despite their widespread use, training deep Transformers can be unstable. Layer normalization, a standard component, improves training stability, but its placement has often been a…

cs.LG2025

Taylor-Model Physics-Informed Neural Networks (PINNs) for Ordinary Differential Equations

Chandra Kanth Nagesh, Sriram Sankaranarayanan, Ramneet Kaur +2

We study the problem of learning neural network models for Ordinary Differential Equations (ODEs) with parametric uncertainties. Such neural network models capture the solution to…