← Search

Kun Gai

52 accepted papers

2026

$\alpha$Matte4K & $\mu$Matting: Dataset and Model for Ultra-Micro Precision Alpha Video Matting

CVPR 2026

High-resolution human video matting aims to predict accurate alpha mattes for semi-transparent regions while ensuring temporal consistency across frames. Despite notable progress, current methods still fail to achieve a satisfactory trade-off between quality and efficiency, with limitations in subje

Cited by 0SourcecodeScholar
2026

3D-Aware Implicit Motion Control for View-Adaptive Human Video Generation

CVPR 2026

Existing methods for human motion control in video generation typically rely on either 2D poses or explicit 3D parametric models (e.g., SMPL) as control signals. However, 2D poses rigidly bind motion to the driving viewpoint, precluding novel-view synthesis. Explicit 3D models, though structurally i

Cited by 0SourcecodeScholar
2026

ALM-MTA: Front-Door Causal Multi-Touch Attribution Method for Creator-Ecosystem Optimization

ICLR 2026poster

Consumption‑Drives‑Production (CDP) on social platforms aims to deliver interpretable incentive signals for creator‑ecosystem building and resource utilization improvement, which strongly relies on attributions. In large-scale and complex recommendation system, the absence of accurate labels togethe…

Cited by 0SourceScholar
2026

Attention as a Compass: Efficient Exploration for Process-Supervised RL in Reasoning Models

ICLR 2026poster

Reinforcement Learning (RL) has shown remarkable success in enhancing the reasoning capabilities of Large Language Models (LLMs). Process-Supervised RL (PSRL) has emerged as a more effective paradigm compared to outcome-based RL. However, existing PSRL approaches suffer from limited exploration effi…

Cited by 0SourcecodeScholar
2026

Boosting Resolution Generalization of Diffusion Transformers with Randomized Positional Encodings

AAAI 2026technical

Resolution generalization in image generation tasks enables the production of higher-resolution images with lower training resolution overhead. However, a key obstacle for diffusion transformers in addressing this problem is the mismatch between positional encodings seen at inference and those used

Cited by 0SourcePDFScholar
2026

Denoising Neural Reranker for Recommender Systems

ICLR 2026poster

For multi-stage recommenders in industry, a user request would first trigger a simple and efficient retriever module that selects and ranks a list of relevant items, then the recommender calls a slower but more sophisticated reranking model that refines the item list exposure to the user. To consist…

Cited by 0SourcecodeScholar
2026

FilmWeaver: Weaving Consistent Multi-Shot Videos with Cache-Guided Autoregressive Diffusion

AAAI 2026technical

Current video generation models perform well at single-shot synthesis but struggle with multi-shot videos, facing critical challenges in maintaining character and background consistency across shots and flexibly generating videos of arbitrary length and shot count. To address these limitations, we i

Cited by 0SourcePDFScholar
2026

GoalRank: Group-Relative Optimization for a Large Ranking Model

ICLR 2026poster

Mainstream ranking approaches typically follow a Generator–Evaluator two-stage paradigm, where a generator produces candidate lists and an evaluator selects the best one. Recent work has attempted to enhance performance by expanding the number of candidate lists, for example, through multi-generator…

Cited by 0SourcecodeScholar
2026

Improving Autoregressive Video Modeling with History Understanding

ICLR 2026poster

Video autoregressive generation (VideoAR) sequentially predicts future frames conditioned on history frames. Despite the advance of recent diffusion-based VideoAR, the role of conditioning signal—internal representations of history frames—remains underexplored. Inspired by the success of strong cond…

Cited by 0SourceScholar
2026

LLM-Aligned Geographic Item Tokenization for Local-Life Recommendation

AAAI 2026technical

Recent advances in Large Language Models (LLMs) have enhanced text-based recommendation by enriching traditional ID-based methods with semantic generalization capabilities. Text-based methods typically encode item textual information via prompt design and generate discrete semantic IDs through item

Cited by 0SourcePDFScholar
2026

Mitigating Noise Shift in Denoising Generative Models with Noise Awareness Guidance

ICLR 2026poster

Existing denoising generative models rely on solving discretized reverse-time SDEs or ODEs. In this paper, we identify a long-overlooked yet pervasive issue in this family of models: a misalignment between the pre-defined noise level and the actual noise level encoded in intermediate states during s…

Cited by 0SourcecodeScholar
2026

Monet: Reasoning in Latent Visual Space Beyond Image and Language

CVPR 2026

Thinking with images has emerged as an effective paradigm for advancing visual reasoning, extending beyond text-only chains of thought by injecting visual evidence into intermediate reasoning steps. However, existing methods fall short of human-like abstract visual thinking, as their flexibility is

Cited by 0SourcecodeScholar
2026

MultiShotMaster: A Controllable Multi-Shot Video Generation Framework

CVPR 2026

Current video generation techniques excel at single-shot clips but struggle to produce narrative multi-shot videos, which require flexible shot arrangement, coherent narrative, and controllability beyond text prompts. To tackle these challenges, we propose MultiShotMaster, a framework for highly con

Cited by 0SourcecodeScholar
2026

Native Spatio-Temporal 4D Variational Autoencoder

ICML 2026poster

Dynamic 3D content representation is crucial for generating moving 3D objects and scenes. Existing 4D variational autoencoders (VAEs) are mainly based on projected 2D pointmaps, which are only incomplete and view-dependent observations that do not model the native 4D positional relations between poi…

Cited by 0SourceScholar
2026

OmniDenseCap: Scripting Multi-Scene Videos with Time-Aware and Structural Audio-Visual Captions

ICML 2026poster

This paper proposes Omni Dense Captioning, a novel task designed to generate continuous, fine-grained, and structured audio-visual narratives with explicit timestamps. To ensure dense semantic coverage, we introduce a six-dimensional structural schema to create "script-like" captions, enabling reade…

Cited by 0SourceScholar
2026

OneSearch: A Preliminary Exploration of the Unified End-to-End Generative Framework for E-commerce Search

ICML 2026poster

Traditional e-commerce search systems employ multi-stage cascading architectures (MCA) that suffer from fragmented computation and optimization objective collisions across stages, ultimately limiting their performance ceiling. We propose OneSearch, the first industrial-deployed end-to-end generative…

Cited by 0SourceScholar
2026

Principled RL for Flow Matching Emerges From the Chunk-level Policy Optimization

ICML 2026poster

Recent Progress in post-training flow matching for text-to-image (T2I) generation with Group Relative Policy Optimization (GRPO) has demonstrated strong potential. However, it is hindered by a critical limitation: inaccurate advantage attribution. In this work, we argue that aggregating consecutive …

Cited by 0SourceScholar
2026

SpatialDiff: 3D-Aware Object Movement via Implicit Spatial Modeling

CVPR 2026

Recent advances in image editing allow impressive manipulation of objects, existing methods still struggle to handle spatial movement in complex scenes, such as objects span different depth layers or are partially occluded. Most image editing methods focus solely on prior information from 2D dataset

Cited by 0SourceScholar
2026

TexEditor: Structure-Preserving Text-Driven texture Editing

ICML 2026poster

Text-guided texture editing aims to modify object appearance while preserving the underlying geometric structure. However, our empirical analysis reveals that even SOTA editing models frequently struggle to maintain structural consistency during texture editing, despite the intended changes being pu…

Cited by 0SourceScholar
2026

VOGUE: Unified Understanding, Generation, and Editing for Videos

ICLR 2026poster

Unified multimodal understanding–generation models have shown promising results in image generation and editing, but remain largely constrained to the image domain. In this work, we present VOGUE, a versatile framework that extends unified modeling to the video domain. VOGUE adopts a dual-stream des…

Cited by 0SourcecodeScholar
2026

VQRAE: Representation Quantization Autoencoders for Multimodal Understanding, Generation and Reconstruction

CVPR 2026

Unifying multimodal understanding, generation and reconstruction representation in a single tokenizer remains a key challenge in building unified models. Previous research predominantly attempts to address this in a dual encoder paradigm, e.g., utilizing the separate encoders for understanding and g

Cited by 0SourcecodeScholar
2026

Visual-Aware CoT: Achieving High-Fidelity Visual Consistency in Unified Models

CVPR 2026

Recently, the introduction of Chain-of-Thought (CoT) has largely improved generation ability of unified models. However, it is observed that the current thinking process during generation mainly focuses on the text consistency with the text prompt, ignoring the visual context consistency with the vi

Cited by 0SourceScholar
2025

DSMoE: Matrix-Partitioned Experts with Dynamic Routing for Computation-Efficient Dense LLMs

EMNLP 2025

As large language models continue to scale, computational costs and resource consumption have emerged as significant challenges. While existing sparsification methods like pruning reduce computational overhead, they risk losing model knowledge through parameter removal. This paper proposes DSMoE (Dy

Cited by 0SourcePDFScholar
2025

Fading to Grow: Growing Preference Ratios via Preference Fading Discrete Diffusion for Recommendation

NeurIPS 2025poster

Recommenders aim to rank items from a discrete item corpus in line with user interests, yet suffer from extremely sparse user preference data. Recent advances in diffusion models have inspired diffusion-based recommenders, which alleviate sparsity by injecting noise during a forward process to preve…

Cited by 0SourceScholar
2025

FullDiT: Video Generative Foundation Models with Multimodal Control via Full Attention

ICCV 2025poster

Current video generative foundation models primarily focus on text-to-video tasks, providing limited control for fine-grained video content creation. Although adapter-based approaches (e.g., ControlNet) enable additional controls with minimal fine-tuning, they encounter challenges when integrating m…

Cited by 0SourcePDFScholar
2025

Granularity-Adaptive Spatial Evidence Tokenization for Video Question Answering

AAAI 2025technical

Video question answering plays a vital role in computer vision, and recent advances in large language models have further propelled the development of this field. However, existing video question answering techniques often face limitations in grasping fine-grained video content in spatial dimensions…

Cited by 0SourcePDFScholar
2025

Imbalance in Balance: Online Concept Balancing in Generation Models

ICCV 2025accepted

In visual generation tasks, the responses and combinations of complex concepts often lack stability and are error-prone, which remains an under-explored area. In this paper, we attempt to explore the causal factors for poor concept responses through elaborately designed experiments. We also design a…

Cited by 0SourcePDFScholar
2025

Improving Video Generation with Human Feedback

NeurIPS 2025poster

Video generation has achieved significant advances through rectified flow techniques, but issues like unsmooth motion and misalignment between videos and prompts persist. In this work, we develop a systematic pipeline that harnesses human feedback to mitigate these problems and refine the video gene…

Cited by 0SourceScholar
2025

LEARN: Knowledge Adaptation from Large Language Model to Recommendation for Practical Industrial Application

AAAI 2025technical

Contemporary recommendation systems predominantly rely on ID embedding to capture latent associations among users and items. However, this approach overlooks the wealth of semantic information embedded within textual descriptions of items, leading to suboptimal performance and poor generalizations.…

2025

Learning Time-Aware Causal Representation for Model Generalization in Evolving Domains

ICML 2025poster

Endowing deep models with the ability to generalize in dynamic scenarios is of vital significance for real-world deployment, given the continuous and complex changes in data distribution. Recently, evolving domain generalization (EDG) has emerged to address distribution shifts over time, aiming to c…

Cited by 0SourcePDFScholar
2025

Making Transformer Decoders Better Differentiable Indexers

ICLR 2025poster

Retrieval aims to find the top-k items most relevant to a query/user from a large dataset. Traditional retrieval models represent queries/users and items as embedding vectors and use Approximate Nearest Neighbor (ANN) search for retrieval. Recently, researchers have proposed a generative-based retri…

Cited by 0SourcePDFScholar
2025

RecFlow: An Industrial Full Flow Recommendation Dataset

ICLR 2025poster

Industrial recommendation systems (RS) rely on the multi-stage pipeline to balance effectiveness and efficiency when delivering items from a vast corpus to users. Existing RS benchmark datasets primarily focus on the exposure space, where novel RS algorithms are trained and evaluated. However, when…

2025

Towards Precise Scaling Laws for Video Diffusion Transformers

CVPR 2025poster

Achieving optimal performance of video diffusion transformers within given data and compute budget is crucial due to their high training costs. This necessitates precisely determining the optimal model size and training hyperparameters before large-scale training. While scaling laws are employed in…

Cited by 3SourcePDFScholar
2025

Who You Are Matters: Bridging Interests and Social Roles via LLM-Enhanced Logic Recommendation

NeurIPS 2025poster

Recommender systems filter contents/items valuable to users by inferring preferences from user features and historical behaviors. Mainstream approaches follow the learning-to-rank paradigm, which focus on discovering and modeling item topics (e.g., categories), and capturing user preferences on the…

Cited by 0SourcecodeScholar
2024

Decoding at the Speed of Thought: Harnessing Parallel Decoding of Lexical Units for LLMs

COLING 2024main

Large language models have demonstrated exceptional capability in natural language understanding and generation. However, their generation speed is limited by the inherently sequential nature of their decoding process, posing challenges for real-time applications. This paper introduces Lexical Unit…

2024

DialogBench: Evaluating LLMs as Human-like Dialogue Systems

NAACL 2024long

Large language models (LLMs) have achieved remarkable breakthroughs in new dialogue capabilities by leveraging instruction tuning,which refreshes human impressions of dialogue systems. The long-standing goal of dialogue systems is to be human-like enough to establish long-term connections with users…

2024

Inductive-Deductive Strategy Reuse for Multi-Turn Instructional Dialogues

EMNLP 2024main

Aligning large language models (LLMs) with human expectations requires high-quality instructional dialogues, which can be achieved by raising diverse, in-depth, and insightful instructions that deepen interactions. Existing methods target instructions from real instruction dialogues as a learning go…

2024

Just Ask One More Time! Self-Agreement Improves Reasoning of Language Models in (Almost) All Scenarios

ACL 2024findings

Although chain-of-thought (CoT) prompting combined with language models has achieved encouraging results on complex reasoning tasks, the naive greedy decoding used in CoT prompting usually causes the repetitiveness and local optimality. To address this shortcoming, ensemble-optimization tries to obt…

2024

Parrot: Enhancing Multi-Turn Instruction Following for Large Language Models

ACL 2024long

Humans often interact with large language models (LLMs) in multi-turn interaction to obtain desired answers or more information. However, most existing studies overlook the multi-turn instruction following ability of LLMs, in terms of training dataset, training method, and evaluation benchmark. In t…

2024

RectifID: Personalizing Rectified Flow with Anchored Classifier Guidance

NeurIPS 2024poster

Customizing diffusion models to generate identity-preserving images from user-provided reference images is an intriguing new problem. The prevalent approaches typically require training on extensive domain-specific images to achieve identity preservation, which lacks flexibility across different use…

2024

Small Agent Can Also Rock! Empowering Small Language Models as Hallucination Detector

EMNLP 2024main

Hallucination detection is a challenging task for large language models (LLMs), and existing studies heavily rely on powerful closed-source LLMs such as GPT-4. In this paper, we propose an autonomous LLM-based agent framework, called HaluAgent, which enables relatively smaller LLMs (e.g. Baichuan2-C…

2024

Unified Language-Vision Pretraining in LLM with Dynamic Discrete Visual Tokenization

ICLR 2024poster

Recently, the remarkable advance of the Large Language Model (LLM) has inspired researchers to transfer its extraordinary reasoning capability to both vision and language data. However, the prevailing approaches primarily regard the visual input as a prompt and focus exclusively on optimizing the te…

2024

Video-LaVIT: Unified Video-Language Pre-training with Decoupled Visual-Motional Tokenization

ICML 2024oral

In light of recent advances in multimodal Large Language Models (LLMs), there is increasing attention to scaling them from image-text data to more informative real-world videos. Compared to static images, video poses unique challenges for effective large-scale pre-training due to the modeling of its…

2023

ResAct: Reinforcing Long-term Engagement in Sequential Recommendation with Residual Actor

ICLR 2023poster

Long-term engagement is preferred over immediate engagement in sequential recommendation as it directly affects product operational metrics such as daily active users (DAUs) and dwell time. Meanwhile, reinforcement learning (RL) is widely regarded as a promising framework for optimizing long-term en…

Cited by 30SourcePDFScholar
2023

State Regularized Policy Optimization on Data with Dynamics Shift

NeurIPS 2023poster

In many real-world scenarios, Reinforcement Learning (RL) algorithms are trained on data with dynamics shift, i.e., with different underlying environment dynamics. A majority of current methods address such issue by training context encoders to identify environment parameters. Data with dynamics shi…

Cited by 17SourcePDFScholar
2020

Dynamic Knapsack Optimization Towards Efficient Multi-Channel Sequential Advertising

ICML 2020poster

In E-commerce, advertising is essential for merchants to reach their target users. The typical objective is to maximize the advertiser’s cumulative revenue over a period of time under a budget constraint. In real applications, an advertisement (ad) usually needs to be exposed to the same user multip…

Cited by 29SourcePDFScholar
2020

Learning to Accelerate Heuristic Searching for Large-Scale Maximum Weighted b-Matching Problems in Online Advertising

IJCAI 2020poster

Bipartite b-matching is fundamental in algorithm design, and has been widely applied into diverse applications, such as economic markets, labor markets, etc. These practical problems usually exhibit two distinct features: large-scale and dynamic, which requires the matching algorithm to be repeatedl…

Cited by 0SourcePDFScholar
2019

Improve Diverse Text Generation by Self Labeling Conditional Variational Auto Encoder

ICASSP 2019accepted

Diversity plays a vital role in many text generating applications. In recent years, Conditional Variational Auto Encoders (CVAE) have shown promising performances for this task. However, they often encounter the so called KL-Vanishing problem. Pervious works use heuristic methods to avoid KL-vanishi…

Cited by 0SourceScholar
2019

Joint Optimization of Tree-based Index and Deep Model for Recommender Systems

NeurIPS 2019poster

Large-scale industrial recommender systems are usually confronted with computational problems due to the enormous corpus size. To retrieve and recommend the most relevant items to users under response time limits, resorting to an efficient index structure is an effective and practical solution. Th…