← Search

Peihao Wang

45 accepted papers

2026

$\nabla$-Reasoner: LLM Reasoning via Test-Time Gradient Descent in Textual Space

ICLR 2026poster

Scaling inference-time compute for Large Language Models (LLMs) has unlocked unprecedented reasoning capabilities. However, existing inference-time scaling methods typically rely on inefficient and suboptimal discrete search algorithms or trial-and-error prompting to improve the online policy. In th…

Cited by 0SourcecodeScholar
2026

Beyond Test-Time Training: Learning to Reason via Hardware-Efficient Optimal Control

ICML 2026poster

Associative memory has long underpinned the design of sequential models. Beyond recall, humans reason by *projecting future states and selecting goal-directed actions*, a capability that modern language models increasingly require but do not natively encode. While prior work uses reinforcement learn…

Cited by 0SourceScholar
2026

FUSE: FK-Steered Multi-Modal Flow Matching for Efficient Simulation-Based Posterior Estimation

ICML 2026poster

Simulation-Based Inference (SBI) is critical for scientific discovery, with generative models offering a promising path toward efficient inference. However, existing methods struggle with effective multimodal modeling. They often rely on brute-force fusion strategies that ignore the structural dispa…

Cited by 0SourceScholar
2026

Oscillation Inversion: Training-Free Image and Video Enhancement Through Oscillated Latents in Large Flow Models

AAAI 2026technical

We explore the oscillatory behavior observed in inversion methods applied to large-scale flow models, including text-to-image and text-to-video. By employing an augmented fixed-point-inspired iterative approach to invert real-world images, we observe that the solution does not achieve convergence, i

Cited by 0SourcePDFScholar
2026

Revisiting Spectral Representations in Generative Diffusion Models

ICML 2026poster

Diffusion models have shown remarkable performance on diverse generation tasks. Recent work finds that imposing representation alignment on the hidden states of diffusion networks can both facilitate training convergence and enhance sampling quality, yet the mechanism driving this synergy remains in…

Cited by 0SourceScholar
2026

Understanding the Mixture-of-Experts with Nadaraya-Watson Kernel

ICLR 2026poster

Mixture-of-Experts (MoE) has become a cornerstone in recent state-of-the-art large language models (LLMs). Traditionally, MoE relies on $\mathrm{Softmax}$ as the router score function to aggregate expert output, a designed choice that has persisted from the earliest MoE models to modern LLMs, and is…

Cited by 0SourceScholar
2026

VLM-3R: Vision-Language Models Augmented with Instruction-Aligned 3D Reconstruction

CVPR 2026

The rapid advancement of Large Multimodal Models (LMMs) for 2D images and videos has sparked interest in extending these models to 3D scenes, with the goal of human-like visual-spatial intelligence. However, achieving deep spatial understanding comparable to human capabilities remains challenging fo

Cited by 0SourcecodeScholar
2026

When Do Graph Foundation Models Transfer? A Data-Centric Theory

ICML 2026poster

Graph foundation models (GFMs) aim to reuse a single backbone across diverse graph domains, yet their transfer is often uneven and can exhibit negative transfer. While most prior work improves transfer through architectural or adaptation choices, we ask a data-centric question: *which properties of …

Cited by 0SourceScholar
2025

CryoFastAR: Fast Cryo-EM Ab initio Reconstruction Made Easy

ICCV 2025poster

Pose estimation from unordered images is fundamental for 3D reconstruction, robotics, and scientific imaging. Recent geometric foundation models, such as DUSt3R, enable end-to-end dense 3D reconstruction but remain underexplored in scientific imaging fields like cryo-electron microscopy (cryo-EM) fo…

Cited by 0SourcePDFScholar
2025

Drag-and-Drop LLMs: Zero-Shot Prompt-to-Weights

NeurIPS 2025poster

Modern Parameter-Efficient Fine-Tuning (PEFT) methods such as low-rank adaptation (LoRA) reduce the cost of customizing large language models (LLMs), yet still require a separate optimization run for every downstream dataset. We introduce \textbf{Drag-and-Drop LLMs (\textit{DnD})}, a prompt-conditio…

Cited by 0SourcecodeScholar
2025

FlexGS: Train Once, Deploy Everywhere with Many-in-One Flexible 3D Gaussian Splatting

CVPR 2025poster

3D Gaussian splatting (3DGS) has enabled various applications in 3D scene representation and novel view synthesis due to its efficient rendering capabilities. However, 3DGS demands significant GPU memory, limiting its use on devices with restricted computational resources. Previous approaches have f…

Cited by 0SourcePDFScholar
2025

Graph-KV: Breaking Sequence via Injecting Structural Biases into Large Language Models

NeurIPS 2025poster

Modern large language models (LLMs) are inherently auto-regressive, requiring input to be serialized into flat sequences regardless of their structural dependencies. This serialization hinders the model’s ability to leverage structural inductive biases, especially in tasks such as retrieval-augmente…

Cited by 0SourceScholar
2025

Rethinking Addressing in Language Models via Contextualized Equivariant Positional Encoding

ICML 2025poster

Transformers rely on both content-based and position-based addressing mechanisms to make predictions, but existing positional encoding techniques often diminish the effectiveness of position-based addressing. Many current methods enforce rigid patterns in attention maps, limiting the ability to mode…

Cited by 1SourcePDFScholar
2025

Steepest Descent Density Control for Compact 3D Gaussian Splatting

CVPR 2025poster

3D Gaussian Splatting (3DGS) has emerged as a powerful technique for real-time, high-resolution novel view synthesis. By representing scenes as a mixture of Gaussian primitives, 3DGS leverages GPU rasterization pipelines for efficient rendering and reconstruction. To optimize scene coverage and capt…

Cited by 0SourcePDFScholar
2025

SteinDreamer: Variance Reduction for Text-to-3D Score Distillation via Stein Identity

AISTATS 2025poster

Score distillation has emerged as one of the most prevalent approaches for text-to-3D asset synthesis. Essentially, score distillation updates 3D parameters by lifting and back-propagating scores averaged over different views. In this paper, we reveal that the gradient estimation in score distillati…

Cited by 0SourceScholar
2025

Understanding and Mitigating Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing

ICLR 2025poster

Structured State Space Models (SSMs) have emerged as alternatives to transformers. While SSMs are often regarded as effective in capturing long-sequence dependencies, we rigorously demonstrate that they are inherently limited by strong recency bias. Our empirical studies also reveal that this bias i…

2024

$\textit{Read-ME}$: Refactorizing LLMs as Router-Decoupled Mixture of Experts with System Co-Design

NeurIPS 2024poster

The proliferation of large language models (LLMs) has led to the adoption of Mixture-of-Experts (MoE) architectures that dynamically leverage specialized subnetworks for improved efficiency and performance. Despite their benefits, MoE models face significant challenges during inference, including in…

2024

$\texttt{Model-GLUE}$: Democratized LLM Scaling for A Large Model Zoo in the Wild

NeurIPS 2024poster

As Large Language Models (LLMs) excel across tasks and specialized domains, scaling LLMs based on existing models has gained significant attention, which is challenged by potential performance drop when combining disparate models. Various techniques have been proposed to aggregate pre-trained LLMs,…

2024

Large Spatial Model: End-to-end Unposed Images to Semantic 3D

NeurIPS 2024poster

Reconstructing and understanding 3D structures from a limited number of images is a classical problem in computer vision. Traditional approaches typically decompose this task into multiple subtasks, involving several stages of complex mappings between different data representations. For example, den…

2024

Lift3D: Zero-Shot Lifting of Any 2D Vision Model to 3D

CVPR 2024poster

In recent years there has been an explosion of 2D vision models for numerous tasks such as semantic segmentation style transfer or scene editing enabled by large-scale 2D image datasets. At the same time there has been renewed interest in 3D scene representations such as neural radiance fields from…

Cited by 6SourcePDFScholar
2024

Polynomial Width is Sufficient for Set Representation with High-dimensional Features

ICLR 2024poster

Set representation has become ubiquitous in deep learning for modeling the inductive bias of neural networks that are insensitive to the input order. DeepSets is the most widely used neural network architecture for set representation. It involves embedding each set element into a latent space with d…

Cited by 6SourcePDFScholar
2024

Taming Mode Collapse in Score Distillation for Text-to-3D Generation

CVPR 2024poster

Despite the remarkable performance of score distillation in text-to-3D generation such techniques notoriously suffer from view inconsistency issues also known as "Janus" artifact where the generated objects fake each view with multiple front faces. Although empirically effective methods have approac…

Cited by 22SourcePDFScholar
2024

VersatileGaussian: Real-time Neural Rendering for Versatile Tasks using Gaussian Splatting

ECCV 2024poster

"The acquisition of multi-task (MT) labels in 3D scenes is crucial for a wide range of real-world applications. Traditional methods generally employ an analysis-by-synthesis approach, generating 2D label maps on novel synthesized views, or utilize Neural Radiance Field (NeRF), which concurrently rep…

Cited by 1SourcePDFScholar
2023

Enhancing NeRF akin to Enhancing LLMs: Generalizable NeRF Transformer with Mixture-of-View-Experts

ICCV 2023poster

Cross-scene generalizable NeRF models, which can directly synthesize novel views of unseen scenes, have become a new spotlight of the NeRF field. Several existing attempts rely on increasingly end-to-end "neuralized" architectures, i.e., replacing scene representation and/or rendering modules with p…

Cited by 21PDFcodeScholar
2023

Equivariant Hypergraph Diffusion Neural Operators

ICLR 2023poster

Hypergraph neural networks (HNNs) using neural networks to encode hypergraphs provide a promising way to model higher-order relations in data and further solve relevant prediction tasks built upon such higher-order relations. However, higher-order relations in practice contain complex patterns and a…

2023

Is Attention All That NeRF Needs?

ICLR 2023poster

We present Generalizable NeRF Transformer (GNT), a transformer-based architecture that reconstructs Neural Radiance Fields (NeRFs) and learns to render novel views on the fly from source views. While prior works on NeRFs optimize a scene representation by inverting a handcrafted rendering equation,…

2023

Learning to Grow Pretrained Models for Efficient Transformer Training

ICLR 2023top-25%

Scaling transformers has led to significant breakthroughs in many domains, leading to a paradigm in which larger versions of existing models are trained and released on a periodic basis. New instances of such models are typically trained completely from scratch, despite the fact that they are often…

Cited by 67SourcePDFScholar
2023

NeRF-SOS: Any-View Self-supervised Object Segmentation on Complex Scenes

ICLR 2023poster

Neural volumetric representations have shown the potential that Multi-layer Perceptrons (MLPs) can be optimized with multi-view calibrated images to represent scene geometry and appearance without explicit 3D supervision. Object segmentation can enrich many downstream applications based on the learn…

2023

NeuralLift-360: Lifting an In-the-Wild 2D Photo to a 3D Object With 360deg Views

CVPR 2023highlight

Virtual reality and augmented reality (XR) bring increasing demand for 3D content generation. However, creating high-quality 3D content requires tedious work from a human expert. In this work, we study the challenging task of lifting a single image to a 3D object and, for the first time, demonstrate…

2023

PINAT: A Permutation INvariance Augmented Transformer for NAS Predictor

AAAI 2023technical

Time-consuming performance evaluation is the bottleneck of traditional Neural Architecture Search (NAS) methods. Predictor-based NAS can speed up performance evaluation by directly predicting performance, rather than training a large number of sub-models and then validating their performance. Most p…

2023

Patch Diffusion: Faster and More Data-Efficient Training of Diffusion Models

NeurIPS 2023poster

Diffusion models are powerful, but they require a lot of time and data to train. We propose Patch Diffusion, a generic patch-wise training framework, to significantly reduce the training time costs while improving data efficiency, which thus helps democratize diffusion model training to broader user…

2022

A Comprehensive Study on Large-Scale Graph Training: Benchmarking and Rethinking

NeurIPS 2022accept

Large-scale graph training is a notoriously challenging problem for graph neural networks (GNNs). Due to the nature of evolving graph structures into the training process, vanilla GNNs usually fail to scale up, limited by the GPU memory space. Up to now, though numerous scalable GNN architectures ha…

2022

Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice

ICLR 2022poster

Vision Transformer (ViT) has recently demonstrated promise in computer vision problems. However, unlike Convolutional Neural Networks (CNN), it is known that the performance of ViT saturates quickly with depth increasing, due to the observed attention collapse or patch uniformity. Despite a couple o…

2022

Aug-NeRF: Training Stronger Neural Radiance Fields With Triple-Level Physically-Grounded Augmentations

CVPR 2022poster

Neural Radiance Field (NeRF) regresses a neural parameterized scene by differentially rendering multi-view images with ground-truth supervision. However, when interpolating novel views, NeRF often yields inconsistent and visually non-smooth geometric results, which we consider as a generalization ga…

Cited by 53PDFcodeScholar
2022

CADTransformer: Panoptic Symbol Spotting Transformer for CAD Drawings

CVPR 2022oral

Understanding 2D computer-aided design (CAD) drawings plays a crucial role for creating 3D prototypes in architecture, engineering and construction (AEC) industries. The task of automated panoptic symbol spotting, i.e., to spot and parse both countable object instances (windows, doors, tables, etc.)…

Cited by 27PDFcodeScholar
2022

Neural Implicit Dictionary Learning via Mixture-of-Expert Training

ICML 2022spotlight

Representing visual signals by coordinate-based deep fully-connected networks has been shown advantageous in fitting complex details and solving inverse problems than discrete grid-based representation. However, acquiring such a continuous Implicit Neural Representation (INR) requires tedious per-sc…

2022

Old can be Gold: Better Gradient Flow can Make Vanilla-GCNs Great Again

NeurIPS 2022accept

Despite the enormous success of Graph Convolutional Networks (GCNs) in modeling graph-structured data, most of the current GCNs are shallow due to the notoriously challenging problems of over-smoothening and information squashing along with conventional difficulty caused by vanishing gradients and o…

2022

Signal Processing for Implicit Neural Representations

NeurIPS 2022accept

Implicit Neural Representations (INRs) encoding continuous multi-media data via multi-layer perceptrons has shown undebatable promise in various computer vision tasks. Despite many successful applications, editing and processing an INR remains intractable as signals are represented by latent paramet…

Cited by 48SourcePDFScholar
2022

SinNeRF: Training Neural Radiance Fields on Complex Scenes from a Single Image

ECCV 2022poster

"Despite the rapid development of Neural Radiance Field (NeRF), the necessity of dense covers largely prohibits its wider applications. While several recent works have attempted to address this issue, they either operate with sparse views (yet still, a few of them) or on simple objects/scenes. In th…

2022

Unified Implicit Neural Stylization

ECCV 2022poster

"Representing visual signals by implicit neural representation (INR) has prevailed among many vision tasks. Its potential for editing/processing given signals remains less explored. This work explores a new intriguing direction: training a stylized implicit representation, using a generalized approa…

2021

Delayed Propagation Transformer: A Universal Computation Engine towards Practical Control in Cyber-Physical Systems

NeurIPS 2021poster

Multi-agent control is a central theme in the Cyber-Physical Systems (CPS). However, current control methods either receive non-Markovian states due to insufficient sensing and decentralized design, or suffer from poor convergence. This paper presents the Delayed Propagation Transformer (DePT), a ne…