← Search

Naichen Shi

5 accepted papers

2026

SURGE:Unbiased Data Assimilation for Diffusion Model via Particle Filtering

ICML 2026poster

Data assimilation (DA) addresses the problem of sequentially estimating the state of a dynamical system from noisy and incomplete observations. In this work, we employ a diffusion model as a world model to simulate and predict the system’s dynamics. Recently, score-based diffusion models have learne…

Cited by 0SourceScholar
2025

Inv-Entropy: A Fully Probabilistic Framework for Uncertainty Quantification in Language Models

NeurIPS 2025poster

Large language models (LLMs) have transformed natural language processing, but their reliable deployment requires effective uncertainty quantification (UQ). Existing UQ methods are often heuristic and lack a fully probabilistic foundation. This paper begins by providing a theoretical justification f…

Cited by 0SourcecodeScholar
2023

Personalized Dictionary Learning for Heterogeneous Datasets

NeurIPS 2023poster

We introduce a relevant yet challenging problem named Personalized Dictionary Learning (PerDL), where the goal is to learn sparse linear representations from heterogeneous datasets that share some commonality. In PerDL, we model each dataset's shared and unique features as global and local dictionar…

Cited by 8SourcePDFScholar
2022

Adam Can Converge Without Any Modification On Update Rules

NeurIPS 2022accept

Ever since \citet{reddi2019convergence} pointed out the divergence issue of Adam, many new variants have been designed to obtain convergence. However, vanilla Adam remains exceptionally popular and it works well in practice. Why is there a gap between theory and practice? We point out there is a mis…

Cited by 98SourcePDFScholar