← Search

Yi Xin

24 accepted papers

2026

Adversarial Reinforcement Learning for Robust Diffusion Large Language Model Unlearning

ICML 2026poster

Diffusion language models (DLMs) have recently emerged as an alternative to autoregressive approaches, enabling parallel sequence generation and flexible token generation orders. Machine unlearning plays a critical role in mitigating legal and ethical risks by removing the influence of specific trai…

Cited by 0SourceScholar
2026

Benchmarking Multimodal Knowledge Conflict for Large Multimodal Models

AAAI 2026technical

Large Multimodal Models (LMMs) face notable challenges when encountering multimodal knowledge conflicts, particularly under retrieval-augmented generation (RAG) frameworks, where the contextual information from external sources may contradict the model’s internal parametric knowledge, leading to unr

Cited by 0SourcePDFScholar
2026

Beyond Conservation: Flexible Molecular Assembly with Unbalanced Diffusion Bridge

AAAI 2026technical

Molecular assembly (MA) has long been a fundamental task in chemistry and biology, with the potential to create new materials and enable novel functions beyond the molecular scale. However, its vast conformational search space poses substantial challenges, and current generative models remain limite

Cited by 0SourcePDFScholar
2026

Efficient Test-Time Scaling via Hierarchical Search and Self-Verification for Discrete Diffusion Language Models

ICML 2026poster

Inference-time compute has re-emerged as a practical way to improve LLM reasoning. Most test-time scaling (TTS) algorithms rely on autoregressive decoding, which is ill-suited to discrete diffusion language models (dLLMs) due to their parallel decoding over the entire sequence. As a result, developi…

Cited by 0SourceScholar
2026

LayerT2V: A Unified Multi-Layer Video Generation Framework

ICML 2026poster

Text-to-video generation has advanced rapidly, but existing methods typically output only the final composited video and lack editable layered representations, limiting their use in professional workflows. We propose \textbf{LayerT2V}, a unified multi-layer video generation framework that produces m…

Cited by 2SourceScholar
2026

MICE-Bench: A Challenging and Comprehensive Benchmark for Multi-Reference Image Creation and Editing

ICML 2026poster

The paradigm of visual generation is rapidly shifting from single-image conditioning toward multi-image conditioning, making the ability to synthesize and edit images based on multiple visual references a critical capability. Despite this trend, existing benchmarks remain largely limited to single-r…

Cited by 0SourceScholar
2026

Rethinking the Design Space of Reinforcement Learning for Diffusion Models: On the Importance of Likelihood Estimation Beyond Loss Design

ICML 2026poster

Reinforcement learning has been widely applied to diffusion and flow models for visual tasks such as text-to-image generation. However, these tasks remain challenging because diffusion models have intractable likelihoods, which creates a barrier for directly applying popular policy-gradient type met…

Cited by 0SourceScholar
2026

TIDE: Temporal-Aware Sparse Autoencoders for Interpretable Diffusion Transformers in Image Generation

AAAI 2026technical

Diffusion Transformers (DiTs) are a powerful yet underexplored class of generative models compared to U-Net-based diffusion architectures. We propose TIDE—Temporal-aware sparse autoencoders for Interpretable Diffusion transformErs—a framework designed to extract sparse, interpretable activation feat

Cited by 0SourcePDFScholar
2026

Threshold-Guided Optimization for Visual Generative Models

ICML 2026poster

Aligning large visual generative models with human feedback is often performed through pairwise preference optimization. While such approaches are conceptually simple, they fundamentally rely on annotated pairs, limiting scalability in settings where feedback is collected as independent scalar ratin…

Cited by 0SourceScholar
2026

UniMedVL: Unifying Medical Multimodal Understanding and Generation through Observation-Knowledge-Analysis

ICML 2026poster

Medical diagnosis demands models that can process multimodal medical inputs, such as medical images and patient histories, and generate diverse outputs including textual reports and visual content, such as annotations or segmentation masks. Despite this need, existing medical AI models disrupt this …

Cited by 0SourceScholar
2026

UniPercept: Towards Unified Perceptual-Level Image Understanding across Aesthetics, Quality, Structure, and Texture

ICML 2026spotlight

Multimodal large language models (MLLMs) have achieved remarkable progress in visual understanding tasks such as visual grounding, segmentation, and captioning. However, their ability to perceive perceptual-level image features remains limited. In this work, we present UniPercept-Bench, a unified fr…

Cited by 0SourceScholar
2026

dMLLM-TTS: Self-Verified and Efficient Test-Time Scaling for Diffusion Multi-Modal Large Language Models

CVPR 2026

Diffusion Multi-modal Large Language Models (dMLLMs) have recently emerged as a novel architecture unifying image generation and understanding. However, developing effective and efficient Test-Time Scaling (TTS) methods to unlock their full generative potential remains an underexplored challenge. To

Cited by 0SourcecodeScholar
2025

$\text{D}_{2}\text{O}$: Dynamic Discriminative Operations for Efficient Long-Context Inference of Large Language Models

ICLR 2025poster

Efficient generative inference in Large Language Models (LLMs) is impeded by the growing memory demands of Key-Value (KV) cache, especially for longer sequences. Traditional KV Cache eviction strategies, which discard less critical KV-pairs based on attention scores, often degrade generation quality…

Cited by 0SourcePDFScholar
2025

Exploit Your Latents: Coarse-Grained Protein Backmapping with Latent Diffusion Models

AAAI 2025technical

Coarse-grained (CG) molecular dynamics of proteins is a preferred approach to studying large molecules on extended time scales by condensing the entire atomic model into a limited number of pseudo-atoms and preserving the thermodynamic properties of the system. However, the significantly increased e…

Cited by 0SourcePDFScholar
2025

From Reflection to Perfection: Scaling Inference-Time Optimization for Text-to-Image Diffusion Models via Reflection Tuning

ICCV 2025poster

Recent text-to-image diffusion models achieve impressive visual quality through extensive scaling of training data and model parameters, yet they often struggle with complex scenes and fine-grained details. Inspired by the self-reflection capabilities emergent in large language models, we propose Re…

2025

Knowledge Is Powerful: Art Knowledge-Driven Framework for Painting Style Classification Integrating Multimodal Knowledge

ICASSP 2025accepted

Paintings possess profound cultural and historical backgrounds. Unlike real-life images, they convey complex semantics beyond simple visual features. This diversity and complexity make painting style classification highly challenging, and many popular visual models struggle with it. To address this…

Cited by 0SourceScholar
2025

Lumina-Image 2.0: A Unified and Efficient Image Generative Framework

ICCV 2025poster

We introduce Lumina-Image 2.0, an advanced text-to-image (T2I) model that surpasses previous state-of-the-art methods across multiple benchmarks. Lumina-Image 2.0 is characterized by two key features: (1) Unification - it adopts a unified architecture (Unified Next-DiT) that treats text and image to…

2025

Robust Logit Adjustment for Learning with Long-Tailed Noisy Data

AAAI 2025technical

Learning with noisy labels (LNL) methods have enabled the deployment of machine learning systems with imperfectly labeled data. However, these methods often struggle to identify noise in the presence of long-tailed (LT) class distributions, where the memorization effect becomes class-dependent. Conv…

Cited by 0SourcePDFScholar
2025

SRPO: Enhancing Multimodal LLM Reasoning via Reflection-Aware Reinforcement Learning

NeurIPS 2025poster

Multimodal large language models (MLLMs) have shown promising capabilities in reasoning tasks, yet still struggle significantly with complex problems requiring explicit self-reflection and self-correction, especially compared to their unimodal text-based counterparts. Existing reflection methods are…

Cited by 0SourceScholar
2025

TR-PTS: Task-Relevant Parameter and Token Selection for Efficient Tuning

ICCV 2025poster

Large pre-trained models achieve remarkable performance in vision tasks but are impractical for fine-tuning due to high computational and storage costs. Parameter-Efficient Fine-Tuning (PEFT) methods mitigate this issue by updating only a subset of parameters; however, most existing approaches are t…

2024

MmAP: Multi-Modal Alignment Prompt for Cross-Domain Multi-Task Learning

AAAI 2024technical

Multi-Task Learning (MTL) is designed to train multiple correlated tasks simultaneously, thereby enhancing the performance of individual tasks. Typically, a multi-task network structure consists of a shared backbone and task-specific decoders. However, the complexity of the decoders increases with t…

Cited by 69SourcePDFScholar
2024

Towards Understanding the Working Mechanism of Text-to-Image Diffusion Model

NeurIPS 2024poster

Recently, the strong latent Diffusion Probabilistic Model (DPM) has been applied to high-quality Text-to-Image (T2I) generation (e.g., Stable Diffusion), by injecting the encoded target text prompt into the gradually denoised diffusion image generator. Despite the success of DPM in practice, the mec…

Cited by 10SourcePDFScholar
2024

V-PETL Bench: A Unified Visual Parameter-Efficient Transfer Learning Benchmark

NeurIPS 2024poster

Parameter-efficient transfer learning (PETL) methods show promise in adapting a pre-trained model to various downstream tasks while training only a few parameters. In the computer vision (CV) domain, numerous PETL algorithms have been proposed, but their direct employment or comparison remains incon…

Cited by 12SourcePDFScholar
2024

VMT-Adapter: Parameter-Efficient Transfer Learning for Multi-Task Dense Scene Understanding

AAAI 2024technical

Large-scale pre-trained models have achieved remarkable success in various computer vision tasks. A standard approach to leverage these models is to fine-tune all model parameters for downstream tasks, which poses challenges in terms of computational and storage costs. Recently, inspired by Natural…

Cited by 63SourcePDFScholar