← Search

Jiasheng Tang

14 accepted papers

2026

Flow Along the $K$-Amplitude for Generative Modeling

ICLR 2026poster

In this work, we propose K-Flow, a novel generative learning paradigm that flows along the $K$-amplitude domain, where $K$ is a scaling parameter that organizes projected coefficients (frequency bands), and amplitude refers to the norm of such coefficients. We instantiate K-Flow with three concrete…

Cited by 0SourcecodeScholar
2026

Lumos-1: On Autoregressive Video Generation with Discrete Diffusion from a Unified Model Perspective

ICLR 2026poster

Autoregressive large language models (LLMs) have unified a vast range of language tasks, inspiring preliminary efforts in autoregressive (AR) video generation. Existing AR video generators either diverge from standard LLM architectures, depend on bulky external text encoders, or incur prohibitive la…

Cited by 0SourcecodeScholar
2026

RAPID$^3$: Tri-Level Reinforced Acceleration Policies for Diffusion Transformer

ICLR 2026poster

Diffusion Transformers (DiTs) excel at visual generation yet remain hampered by slow sampling. Existing training-free accelerators—step reduction, feature caching, and sparse attention—enhance inference speed but typically rely on a uniform heuristic or manually designed adaptive strategy for all i…

Cited by 0SourceScholar
2025

A Stitch in Time Saves Nine: Small VLM is a Precise Guidance for Accelerating Large VLMs

CVPR 2025poster

Vision-language models (VLMs) have shown remarkable success across various multi-modal tasks, yet large VLMs encounter significant efficiency challenges due to processing numerous visual tokens. A promising approach to accelerating large VLM inference is using partial information, such as attention…

2025

Dynamic Diffusion Transformer

ICLR 2025poster

Diffusion Transformer (DiT), an emerging diffusion model for image generation, has demonstrated superior performance but suffers from substantial computational costs. Our investigations reveal that these costs stem from the static inference paradigm, which inevitably introduces redundant computation…

2025

FPSAttention: Training-Aware FP8 and Sparsity Co-Design for Fast Video Diffusion

NeurIPS 2025spotlight

Diffusion generative models have become the standard for producing high-quality, coherent video content, yet their slow inference speeds and high computational demands hinder practical deployment. Although both quantization and sparsity can independently accelerate inference while maintaining genera…

Cited by 0SourceScholar
2025

Inference-Time Alignment of Diffusion Models with Direct Noise Optimization

ICML 2025poster

In this work, we focus on the alignment problem of diffusion models with a continuous reward function, which represents specific objectives for downstream tasks, such as increasing darkness or improving the aesthetics of images. The central goal of the alignment problem is to adjust the distribution…

Cited by 0SourcePDFScholar
2025

Poison as Cure: Visual Noise for Mitigating Object Hallucinations in LVMs

NeurIPS 2025poster

Large vision-language models (LVMs) extend large language models (LLMs) with visual perception capabilities, enabling them to process and interpret visual information. A major challenge compromising their reliability is object hallucination that LVMs may generate plausible but factually inaccurate i…

Cited by 0SourceScholar
2025

SparseDiT: Token Sparsification for Efficient Diffusion Transformer

NeurIPS 2025poster

Diffusion Transformers (DiT) are renowned for their impressive generative performance; however, they are significantly constrained by considerable computational costs due to the quadratic complexity in self-attention and the extensive sampling steps required. While advancements have been made in exp…

Cited by 0SourcecodeScholar
2024

Accelerating Parallel Sampling of Diffusion Models

ICML 2024poster

Diffusion models have emerged as state-of-the-art generative models for image generation. However, sampling from diffusion models is usually time-consuming due to the inherent autoregressive nature of their sampling process. In this work, we propose a novel approach that accelerates the sampling of…

2024

Dynamic Tuning Towards Parameter and Inference Efficiency for ViT Adaptation

NeurIPS 2024poster

Existing parameter-efficient fine-tuning (PEFT) methods have achieved significant success on vision transformers (ViTs) adaptation by improving parameter efficiency. However, the exploration of enhancing inference efficiency during adaptation remains underexplored. This limits the broader applicatio…

2024

ShapeBoost: Boosting Human Shape Estimation with Part-Based Parameterization and Clothing-Preserving Augmentation

AAAI 2024technical

Accurate human shape recovery from a monocular RGB image is a challenging task because humans come in different shapes and sizes and wear different clothes. In this paper, we propose ShapeBoost, a new human shape recovery framework that achieves pixel-level alignment even for rare body shapes and hi…

Cited by 2SourcePDFScholar
2023

NIKI: Neural Inverse Kinematics With Invertible Neural Networks for 3D Human Pose and Shape Estimation

CVPR 2023poster

With the progress of 3D human pose and shape estimation, state-of-the-art methods can either be robust to occlusions or obtain pixel-aligned accuracy in non-occlusion cases. However, they cannot obtain robustness and mesh-image alignment at the same time. In this work, we present NIKI (Neural Invers…