← Search

Wenlei Shi

4 accepted papers

2026

Accordion-Thinking: Self-Regulated Step Summaries for Efficient and Readable LLM Reasoning

ICML 2026poster

Scaling test-time compute via long Chain-of-Thought unlocks remarkable gains in reasoning capabilities, yet it faces practical limits due to the linear growth of KV cache and quadratic attention complexity. In this paper, we introduce AccordionThinking, an end-to-end framework where LLMs learn to se…

Cited by 0SourceScholar
2025

Flaming-hot Initiation with Regular Execution Sampling for Large Language Models

NAACL 2025findings

Since the release of ChatGPT, large language models (LLMs) have demonstrated remarkable capabilities across various domains. A key challenge in developing these general capabilities is efficiently sourcing diverse, high-quality data. This becomes especially critical in reasoning-related tasks with s…

Cited by 2SourcePDFScholar
2023

Learning Physics-Informed Neural Networks without Stacked Back-propagation

AISTATS 2023poster

Physics-Informed Neural Network (PINN) has become a commonly used machine learning approach to solve partial differential equations (PDE). But, facing high-dimensional secondorder PDE problems, PINN will suffer from severe scalability issues since its loss includes second-order derivatives, the comp…

2023

NeuralStagger: Accelerating Physics-constrained Neural PDE Solver with Spatial-temporal Decomposition

ICML 2023poster

Neural networks have shown great potential in accelerating the solution of partial differential equations (PDEs). Recently, there has been a growing interest in introducing physics constraints into training neural PDE solvers to reduce the use of costly data and improve the generalization ability. H…

Cited by 11SourcePDFScholar