← Search

Lemeng Wu

22 accepted papers

2026

VideoAuto-R1: Video Auto Reasoning via Thinking Once, Answering Twice

CVPR 2026

Chain-of-thought (CoT) reasoning has emerged as a powerful tool for multimodal large language models on video understanding tasks. However, its necessity and advantages over direct answering remain underexplored. In this paper, we first demonstrate that for RL-trained video models, direct answering

Cited by 0SourceScholar
2026

dTRPO : Trajectory Reduction in Policy Optimization of Diffusion Large Language Models

ICML 2026poster

Diffusion Large Language Models (dLLMs) introduce a new paradigm for language generation and thus induce new challenges in aligning dLLMs for human preference. In this work, aim to optimize the dLLM generation process by developing a theoretical formulation and an efficient and effective quantificat…

Cited by 0SourceScholar
2025

EdgeTAM: On-Device Track Anything Model

CVPR 2025poster

On top of Segment Anything Model (SAM), SAM 2 further extends its capability from image to video inputs through a memory bank mechanism and obtains a remarkable performance compared with previous methods, making it a foundation model for video segmentation task. In this paper, we aim at making SAM 2…

2025

LongVU: Spatiotemporal Adaptive Compression for Long Video-Language Understanding

ICML 2025poster

Multimodal Large Language Models (MLLMs) have shown promising progress in understanding and analyzing video content. However, processing long videos remains a significant challenge constrained by LLM's context size. To address this limitation, we propose \textbf{LongVU}, a spatiotemporal adaptive co…

2025

Longhorn: State Space Models are Amortized Online Learners

ICLR 2025poster

The most fundamental capability of modern AI methods such as Large Language Models (LLMs) is the ability to predict the next token in a long sequence of tokens, known as “sequence modeling.” Although the Transformers model is the current dominant approach to sequence modeling, its quadratic computa…

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
2024

Communication Efficient Distributed Training with Distributed Lion

NeurIPS 2024poster

The Lion optimizer has been a promising competitor with the AdamW for training large AI models, with advantages in memory, computation, and sample efficiency. In this paper, we introduce Distributed Lion, an innovative adaptation of Lion for distributed training environments. Leveraging the sign ope…

Cited by 4SourcePDFScholar
2024

EfficientSAM: Leveraged Masked Image Pretraining for Efficient Segment Anything

CVPR 2024highlight

Segment Anything Model (SAM) has emerged as a powerful tool for numerous vision applications. A key component that drives the impressive performance for zero-shot transfer and high versatility is a super large Transformer model trained on the extensive high-quality SA-1B dataset. While beneficial th…

2024

LanguageFlow: Advancing Diffusion Language Generation with Probabilistic Flows

NAACL 2024long

Recent works have demonstrated success in controlling sentence attributes (e.g., sentiment) and structure (e.g., syntactic structure) based on the diffusion language model. A key component that drives theimpressive performance for generating high-quality samples from noise is iteratively denoise for…

Cited by 2SourcePDFScholar
2024

Layer Compression of Deep Networks with Straight Flows

AAAI 2024technical

Very deep neural networks lead to significantly better performance on various real tasks. However, it usually causes slow inference and is hard to be deployed on real-world devices. How to reduce the number of layers to save memory and to accelerate the inference is an eye-catching topic. In thi…

2024

PrivCirNet: Efficient Private Inference via Block Circulant Transformation

NeurIPS 2024poster

Homomorphic encryption (HE)-based deep neural network (DNN) inference protects data and model privacy but suffers from significant computation overhead. We observe transforming the DNN weights into circulant matrices converts general matrix-vector multiplications into HE-friendly 1-dimensional convo…

2023

Fast Point Cloud Generation With Straight Flows

CVPR 2023poster

Diffusion models have emerged as a powerful tool for point cloud generation. A key component that drives the impressive performance for generating high-quality samples from noise is iteratively denoise for thousands of steps. While beneficial, the complexity of learning steps has limited its applica…

2023

FlowGrad: Controlling the Output of Generative ODEs With Gradients

CVPR 2023poster

Generative modeling with ordinary differential equations (ODEs) has achieved fantastic results on a variety of applications. Yet, few works have focused on controlling the generated content of a pre-trained ODE-based generative model. In this paper, we propose to optimize the output of ODE models ac…

2022

Diffusion-based Molecule Generation with Informative Prior Bridges

NeurIPS 2022accept

AI-based molecule generation provides a promising approach to a large area of biomedical sciences and engineering, such as antibody design, hydrolase engineering, or vaccine development. Because the molecules are governed by physical laws, a key challenge is to incorporate prior information into the…

Cited by 119SourcePDFScholar
2022

First Hitting Diffusion Models for Generating Manifold, Graph and Categorical Data

NeurIPS 2022accept

We propose a family of First Hitting Diffusion Models (FHDM), deep generative models that generate data with a diffusion process that terminates at a random first hitting time. This yields an extension of the standard fixed-time diffusion models that terminate at a pre-specified deterministic time.…

Cited by 29SourcePDFScholar
2022

How to Fill the Optimum Set? Population Gradient Descent with Harmless Diversity

ICML 2022spotlight

Although traditional optimization methods focus on finding a single optimal solution, most objective functions in modern machine learning problems, especially those in deep learning, often have multiple or infinite number of optimal points. Therefore, it is useful to consider the problem of finding…

2020

Firefly Neural Architecture Descent: a General Approach for Growing Neural Networks

NeurIPS 2020poster

We propose firefly neural architecture descent, a general framework for progressively and dynamically growing neural networks to jointly optimize the networks' parameters and architectures. Our method works in a steepest descent fashion, which iteratively finds the best network within a functional n…

2020

Greedy Optimization Provably Wins the Lottery: Logarithmic Number of Winning Tickets is Enough

NeurIPS 2020poster

Despite the great success of deep learning, recent works show that large deep neural networks are often highly redundant and can be significantly reduced in size. However, the theoretical question of how much we can prune a neural network given a specified tolerance of accuracy drop is still open. T…