← Search

Xin Geng

87 accepted papers

2026

A Unified Framework for Knowledge Transfer in Bidirectional Model Scaling

CVPR 2026

Transferring pre-trained knowledge from a source model to a target model of a different architectural size is a key challenge for flexible and efficient model scaling. However, current parameter-space methods treat Small-to-Large (S2L) and Large-to-Small (L2S) scaling as separate, incompatible probl

Cited by 0SourceScholar
2026

Adaptive-Learngene: Continual Expansion and Task-Aware Selection of Learngenes for Dynamic Environments

AAAI 2026technical

Pre-trained Vision Transformer (ViT) models have achieved impressive performance across various computer vision tasks. However, most existing pre-trained models are built on fixed datasets and lack the flexibility to incorporate new pre-training data. When additional data becomes available, previous

Cited by 0SourcePDFScholar
2026

Alignment through Meta-Weighted Online Sampling: Bridging the Gap between Data Generation and Preference Optimization

ICLR 2026poster

Preference optimization is crucial for aligning large language models (LLMs) with human values and intentions. A significant challenge in this process is the distribution mismatch between pre-collected offline preference data and the evolving model policy. Existing methods attempt to reduce this gap…

Cited by 0SourcecodeScholar
2026

Breaking Semantic Boundaries: Distribution-Guided Semantic Exploration for Creative Generation

CVPR 2026

Text-to-image (T2I) diffusion models effectively produce semantically aligned images, but their reliance on training distributions constrains their capacity for synthesizing truly novel, out-of-distribution concepts. Existing methods attempt to enhance creativity through semantic exploration, such a

Cited by 0SourceScholar
2026

Breaking the Scale Barrier: One-Shot Knowledge Transfer via Frequency Transform

ICML 2026poster

Transferring knowledge by fine-tuning large-scale pre-trained networks has become a standard paradigm for downstream tasks, yet the knowledge of a pre-trained model is tightly coupled with monolithic architecture, which restricts flexible reuse across models of varying scales. In response to this ch…

Cited by 0SourceScholar
2026

Class-Prior Perturbation-Robust Regularization for Imbalanced Unreliable Partial Label Learning

ICML 2026poster

Imbalanced Unreliable Partial Label Learning (I-UPLL) is a challenging weakly supervised learning setting in which severe class imbalance and unreliable candidate labels jointly degrade model performance. By revisiting existing approaches for imbalanced learning, we observe that most of them fundame…

Cited by 0SourceScholar
2026

DivControl: Knowledge Diversion for Controllable Image Generation

AAAI 2026technical

Diffusion models have advanced from text-to-image (T2I) to image-to-image (I2I) generation by incorporating structured inputs such as depth maps, enabling fine-grained spatial control. However, existing methods either train separate models for each condition or rely on unified architectures with ent

Cited by 0SourcePDFScholar
2026

Extracting Multimodal Learngene in CLIP: Unveiling the Multimodal Generalizable Knowledge

AAAI 2026technical

CLIP (Contrastive Language-Image Pre-training) has attracted widespread attention for its multimodal generalizable knowledge, which is significant for downstream tasks. However, the computational overhead of a large number of parameters and large-scale pre-training poses challenges of pre-training a

Cited by 0SourcePDFScholar
2026

FINE: Factorizing Knowledge for Initialization of Variable-sized Diffusion Models

CVPR 2026

The training of diffusion models is computationally intensive, making effective pre-training essential. However, real-world deployments often demand models of variable sizes due to diverse memory and computational constraints, posing challenges when corresponding pre-trained versions are unavailable

Cited by 0SourceScholar
2026

FedHarmony: Harmonizing Heterogeneous Label Correlations in Federated Multi-Label Learning

CVPR 2026

Federated Multi-Label Learning is a distributed paradigm where multiple clients possess heterogeneous multi-label data and perform collaborative learning under privacy constraints without sharing raw data. However, modeling label correlations under heterogeneous distributions remains challenging. Du

Cited by 0SourceScholar
2026

FedPAT: Federated Test-Time Adaptation via Prototype Affinity Topology

ICML 2026poster

Federated Learning (FL) enables privacy-preserving collaboration among distributed clients in open-world environments, but its performance often degrades under data heterogeneity and unpredictable distribution shifts. Test-Time Adaptation (TTA) has recently been introduced into FL to leverage unlabe…

Cited by 0SourceScholar
2026

Inheriting Generalizable Knowledge from LLMs to Diverse Vertical Tasks

ICLR 2026poster

Large language models (LLMs) have demonstrated remarkable generalization across diverse tasks, suggesting the existence of task-agnostic, generalizable knowledge encoded within them. However, how to systematically extract and evaluate this knowledge remains unexplored. In this work, we innovatively…

Cited by 0SourcecodeScholar
2026

Knowledge Diversion for Efficient Morphology Control and Policy Transfer

ICML 2026poster

Universal morphology control aims to learn a universal policy that generalizes across heterogeneous robot morphologies, with Transformer-based controllers emerging as a dominant choice. However, such architectures incur substantial computational costs, resulting in high deployment overhead, and exis…

Cited by 0SourceScholar
2026

Learngene: Inheritable ‘Genes’ in Intelligent Agents (Abstract Reprint)

AAAI 2026technical

Biological intelligence has driven significant progress in artificial intelligence (AI), but a critical gap remains: biological systems inherit innate abilities from genes, with brains initialized by blueprints refined over 3.5 billion years of evolution, while machines rely heavily on inefficient,

Cited by 0SourcePDFScholar
2026

Positive–Unlabeled Reinforcement Learning Distillation for On-Premise Small Models

ICML 2026poster

Due to constraints on privacy, cost, and latency, on-premise deployment of small models is increasingly common. However, most practical pipelines stop at supervised fine-tuning (SFT) and fail to reach the reinforcement learning (RL) alignment stage. The main reason is that RL alignment typically req…

Cited by 0SourceScholar
2026

RE-TRAC: REcursive TRAjectory Compression for Deep Search Agents

ICML 2026poster

LLM-based deep research agents are largely built on the ReAct framework. This linear design makes it difficult to revisit earlier states, branch into alternative search directions, or maintain global awareness under long contexts, often leading to local optima, redundant exploration, and inefficient…

Cited by 0SourceScholar
2026

Self-Supervised Weight Templates for Scalable Vision Model Initialization

ICML 2026poster

The increasing scale and complexity of modern model parameters underscore the importance of pre-trained models. However, deployment often demands architectures of varying sizes, exposing limitations of conventional pre-training and fine-tuning. To address this, we propose SWEET, a self-supervised fr…

Cited by 0SourceScholar
2026

Towards On-Policy SFT: Distribution Discriminant Theory and its Applications in LLM Training

ICML 2026poster

Supervised fine-tuning (SFT) is computationally efficient but often yields inferior generalization compared to reinforcement learning (RL). This gap is primarily driven by RL’s use of on-policy data. We propose a framework to bridge this chasm by enabling On-Policy SFT. We first present ***Distribut…

Cited by 0SourceScholar
2026

Trustworthy Federated Label Distribution Learning under Annotation Quality Disparity

ICML 2026poster

Label Distribution Learning (LDL) models supervision as an instance-wise probability distribution, enabling fine-grained learning under inherent ambiguity, but its success relies on high-fidelity label distributions that are costly to obtain and thus often noisy. Motivated by privacy-sensitive appli…

Cited by 0SourceScholar
2026

Understanding Performance Collapse in Layer-Pruned Large Language Models via Decision Representation Transitions

ICML 2026poster

Layer pruning efficiently reduces Large Language Model (LLM) computational costs but often triggers sudden performance collapse. Existing representation-based analyses struggle to explain this mechanism. We propose studying pruning through decision representation. Focusing on multiple-choice tasks, …

Cited by 0SourceScholar
2026

Unlocking Pre-trained Weights: Parameter Inheritance for Zero-Shot Initialization

CVPR 2026

Appropriate parameter initialization is crucial for reducing the training cost of deep neural networks. Graph HyperNetworks (GHN) have emerged as a promising approach for initializing diverse architectures, with recent methods such as Task-Aware Learngene (TAL) further attempting to leverage pre-tra

Cited by 0SourcecodeScholar
2026

When Labelers Stay Silent: The Power of Ties in Cost-Effective Preference Learning

ICML 2026poster

Standard preference alignment relies on a binary forced-choice paradigm, assuming definitive preferences for all pairs. However, we find that indistinguishable pairs are prevalent even in standard benchmarks, where quality differences of two responses often fall below the labeler's discriminative re…

Cited by 0SourceScholar
2025

ECO: Evolving Core Knowledge for Efficient Transfer

NeurIPS 2025poster

Knowledge in modern neural networks is often entangled and structurally opaque, making current transfer methods—typically based on reusing entire parameter sets—inefficient and inflexible. Efforts to improve flexibility by reusing partial parameters frequently depend on handcrafted heuristics or rig…

Cited by 0SourceScholar
2025

Fast Large Language Model Collaborative Decoding via Speculation

ICML 2025poster

Large Language Model (LLM) collaborative decoding techniques improve output quality by combining the outputs of multiple models at each generation step, but they incur high computational costs. In this paper, we introduce **Collaborative decoding via Speculation (CoS)**, a novel framework that accel…

2025

FlowPrune: Accelerating Attention Flow Calculation by Pruning Flow Network

NeurIPS 2025poster

The Transformer architecture serves as the foundation of modern AI systems, powering recent advances in Large Language Models (LLMs) and Large Multimodal Models (LMMs). Central to these models, attention mechanisms capture contextual dependencies via token interactions. Beyond inference, attention h…

Cited by 0SourceScholar
2025

Improved Noise Schedule for Diffusion Training

ICCV 2025poster

Diffusion models have emerged as the de facto choice for generating high-quality visual signals across various domains.However, training a single model to predict noise across various levels poses significant challenges, necessitating numerous iterations and incurring significant computational costs…

Cited by 0SourcePDFScholar
2025

Inheriting Generalized Learngene for Efficient Knowledge Transfer across Multiple Tasks

AAAI 2025technical

In practical applications, it is often necessary to transfer knowledge from large pretrained models to small ones with various architectures for tackling different tasks. The Learngene framework, proposed recently, firstly extracts one compact module termed as learngene from a large well-trained mod…

Cited by 0SourcePDFScholar
2025

KIND: Knowledge Integration and Diversion for Training Decomposable Models

ICML 2025poster

Pre-trained models have become the preferred backbone due to the increasing complexity of model parameters. However, traditional pre-trained models often face deployment challenges due to their fixed sizes, and are prone to negative transfer when discrepancies arise between training tasks and target…

2025

Label Distribution Learning with Biased Annotations Assisted by Multi-Label Learning

IJCAI 2025

Multi-label learning (MLL) has gained attention for its ability to represent real-world data. Label Distribution Learning (LDL), an extension of MLL to learning from label distributions, faces challenges in collecting accurate label distributions. To address the issue of biased annotations, based on

Cited by 0SourcePDFScholar
2025

Learngene Tells You How to Customize: Task-Aware Parameter Initialization at Flexible Scales

ICML 2025poster

Appropriate parameter initialization strategies are essential for reducing the high computational costs of training large pretrained models in various task scenarios. Graph HyperNetwork (GHN), a parameter initialization method, has recently demonstrated strong performance in initializing models. How…

Cited by 0SourcePDFScholar
2025

Mimic In-Context Learning for Multimodal Tasks

CVPR 2025poster

Recently, In-context Learning (ICL) has become a significant inference paradigm in Large Multimodal Models (LMMs), utilizing a few in-context demonstrations (ICDs) to prompt LMMs for new tasks. However, the synergistic effects in multimodal data increase the sensitivity of ICL performance to the con…

2025

RankMatch: A Novel Approach to Semi-Supervised Label Distribution Learning Leveraging Rank Correlation between Labels

NeurIPS 2025poster

Pseudo label based semi-supervised learning (SSL) for single-label and multi-label classification tasks has been extensively studied; however, semi-supervised label distribution learning (SSLDL) remains a largely unexplored area. Existing SSL methods fail in SSLDL because the pseudo-labels they ge…

Cited by 0SourceScholar
2025

Redefining <Creative> in Dictionary: Towards an Enhanced Semantic Understanding of Creative Generation

CVPR 2025poster

Creative remains an inherently abstract concept for both humans and diffusion models. While text-to-image (T2I) diffusion models can easily generate out-of-distribution concepts like "a blue banana", they struggle with generating combinatorial objects such as "a creative mixture that resembles a let…

2025

Reduction-based Pseudo-label Generation for Instance-dependent Partial Label Learning

NeurIPS 2025poster

Instance-dependent Partial Label Learning (ID-PLL) aims to learn a multi-class predictive model given training instances annotated with candidate labels related to features, among which correct labels are hidden fixed but unknown. The previous works involve leveraging the identification capability o…

Cited by 0SourceScholar
2025

SoPo: Text-to-Motion Generation Using Semi-Online Preference Optimization

NeurIPS 2025poster

Text-to-motion generation is essential for advancing the creative industry but often presents challenges in producing consistent, realistic motions. To address this, we focus on fine-tuning text-to-motion models to consistently favor high-quality, human-preferred motions—a critical yet largely unexp…

Cited by 0SourcecodeScholar
2025

VADIS: Investigating Inter-View Representation Biases for Multi-View Partial Multi-Label Learning

UAI 2025

Multi-view partial multi-label learning (MVPML) deals with training data where each example is represented by multiple feature vectors and associated with a set of candidate labels, only a subset of which are correct. The diverse representation biases present in different views complicate the annota

Cited by 0SourcePDFScholar
2025

WAVE: Weight Templates for Adaptive Initialization of Variable-sized Models

CVPR 2025poster

The growing complexity of model parameters underscores the significance of pre-trained models. However, deployment constraints often necessitate models of varying sizes, exposing limitations in the conventional pre-training and fine-tuning paradigm, particularly when target model sizes are incompati…

2024

Aligned Objective for Soft-Pseudo-Label Generation in Supervised Learning

ICML 2024poster

Soft pseudo-labels, generated by the softmax predictions of the trained networks, offer a probabilistic rather than binary form, and have been shown to improve the performance of deep neural networks in supervised learning. Most previous methods adopt classification loss to train a classifier as the…

Cited by 1SourcePDFScholar
2024

Aligning Vision Models with Human Aesthetics in Retrieval: Benchmarks and Algorithms

NeurIPS 2024poster

Modern vision models are trained on very large noisy datasets. While these models acquire strong capabilities, they may not follow the user's intent to output the desired results in certain aspects, e.g., visual aesthetic, preferred style, and responsibility. In this paper, we target the realm of vi…

Cited by 3SourcePDFScholar
2024

Building Variable-Sized Models via Learngene Pool

AAAI 2024technical

Recently, Stitchable Neural Networks (SN-Net) is proposed to stitch some pre-trained networks for quickly building numerous networks with different complexity and performance trade-offs. In this way, the burdens of designing or training the variable-sized networks, which can be used in application s…

2024

Cluster-Learngene: Inheriting Adaptive Clusters for Vision Transformers

NeurIPS 2024poster

In recent years, the merging of vast datasets with powerful computational resources has led to the emergence of large pre-trained models in the field of deep learning. However, the common practices often overgeneralize the applicability of these models, overlooking the task-specific resource constra…

Cited by 1SourcePDFScholar
2024

Correlation-Induced Label Prior for Semi-Supervised Multi-Label Learning

ICML 2024poster

Semi-supervised multi-label learning (SSMLL) aims to address the challenge of limited labeled data availability in multi-label learning (MLL) by leveraging unlabeled data to improve the model's performance. Due to the difficulty of estimating the reliable label correlation on minimal multi-labeled d…

Cited by 0SourcePDFScholar
2024

Exploiting Multi-Label Correlation in Label Distribution Learning

IJCAI 2024poster

Label Distribution Learning (LDL) is a novel machine learning paradigm that assigns label distribution to each instance. Numerous LDL methods proposed to leverage label correlation in the learning process to solve the exponential-sized output space; among these, many exploited the low-rank structur…

2024

Exploring Learngene via Stage-wise Weight Sharing for Initializing Variable-sized Models

IJCAI 2024poster

In practice, we usually need to build variable-sized models adapting for diverse resource constraints in different application scenarios, where weight initialization is an important step prior to training. The Learngene framework, introduced recently, firstly learns one compact part termed as learng…

2024

Initializing Variable-sized Vision Transformers from Learngene with Learnable Transformation

NeurIPS 2024poster

In practical scenarios, it is necessary to build variable-sized models to accommodate diverse resource constraints, where weight initialization serves as a crucial step preceding training. The recently introduced Learngene framework firstly learns one compact module, termed learngene, from a large w…

Cited by 4SourcePDFScholar
2024

LIVE: Learnable In-Context Vector for Visual Question Answering

NeurIPS 2024poster

As language models continue to scale, Large Language Models (LLMs) have exhibited emerging capabilities in In-Context Learning (ICL), enabling them to solve language tasks by prefixing a few in-context demonstrations (ICDs) as context. Inspired by these advancements, researchers have extended these…

2024

Learning with Partial-Label and Unlabeled Data: A Uniform Treatment for Supervision Redundancy and Insufficiency

ICML 2024spotlight

One major challenge in weakly supervised learning is learning from inexact supervision, ranging from partial labels (PLs) with *redundant* information to the extreme of unlabeled data with *insufficient* information. While recent work has made significant strides in specific inexact supervision cont…

Cited by 2SourcePDFScholar
2024

Linearly Decomposing and Recomposing Vision Transformers for Diverse-Scale Models

NeurIPS 2024poster

Vision Transformers (ViTs) are widely used in a variety of applications, while they usually have a fixed architecture that may not match the varying computational resources of different deployment environments. Thus, it is necessary to adapt ViT architectures to devices with diverse computational ov…

Cited by 3SourcePDFScholar
2024

Transformer as Linear Expansion of Learngene

AAAI 2024technical

We propose expanding the shared Transformer module to produce and initialize Transformers of varying depths, enabling adaptation to diverse resource constraints. Drawing an analogy to genetic expansibility, we term such module as learngene. To identify the expansion mechanism, we delve into the rela…

2024

ULAREF: A Unified Label Refinement Framework for Learning with Inaccurate Supervision

ICML 2024spotlight

Learning with inaccurate supervision is often encountered in weakly supervised learning, and researchers have invested a considerable amount of time and effort in designing specialized algorithms for different forms of annotations in inaccurate supervision. In fact, different forms of these annotati…

Cited by 0SourcePDFScholar
2024

What Makes Partial-Label Learning Algorithms Effective?

NeurIPS 2024poster

A partial label (PL) specifies a set of candidate labels for an instance and partial-label learning (PLL) trains multi-class classifiers with PLs. Recently, many methods that incorporate techniques from other domains have shown strong potential. The expectation that stronger techniques would enhance…

Cited by 2SourcePDFScholar
2023

Decompositional Generation Process for Instance-Dependent Partial Label Learning

ICLR 2023top-25%

Partial label learning (PLL) is a typical weakly supervised learning problem, where each training example is associated with a set of candidate labels among which only one is true. Most existing PLL approaches assume that the incorrect labels in each training example are randomly picked as the candi…

2023

Efficient Diffusion Training via Min-SNR Weighting Strategy

ICCV 2023poster

Denoising diffusion models have been a mainstream approach for image generation, however, training these models often suffers from slow convergence. In this paper, we discovered that the slow convergence is partly due to conflicting optimization directions between timesteps. To address this issue, w…

Cited by 149PDFcodeScholar
2023

Exploring Diverse In-Context Configurations for Image Captioning

NeurIPS 2023poster

After discovering that Language Models (LMs) can be good in-context few-shot learners, numerous strategies have been proposed to optimize in-context sequence configurations. Recently, researchers in Vision-Language (VL) domains also develop their few-shot learners, while they only use the simplest w…

2023

FREDIS: A Fusion Framework of Refinement and Disambiguation for Unreliable Partial Label Learning

ICML 2023poster

To reduce the difficulty of annotation, partial label learning (PLL) has been widely studied, where each example is ambiguously annotated with a set of candidate labels instead of the exact correct label. PLL assumes that the candidate label set contains the correct label, which induces disambiguati…

Cited by 7SourcePDFScholar
2023

Progressive Purification for Instance-Dependent Partial Label Learning

ICML 2023poster

Partial label learning (PLL) aims to train multiclass classifiers from the examples each annotated with a set of candidate labels where a fixed but unknown candidate label is correct. In the last few years, the instance-independent generation process of candidate labels has been extensively studied,…

Cited by 25SourcePDFScholar
2023

Towards Effective Visual Representations for Partial-Label Learning

CVPR 2023poster

Under partial-label learning (PLL) where, for each training instance, only a set of ambiguous candidate labels containing the unknown true label is accessible, contrastive learning has recently boosted the performance of PLL on vision tasks, attributed to representations learned by contrasting the s…

2022

Ambiguity-Induced Contrastive Learning for Instance-Dependent Partial Label Learning

IJCAI 2022poster

Partial label learning (PLL) learns from a typical weak supervision, where each training instance is labeled with a set of ambiguous candidate labels (CLs) instead of its exact ground-truth label. Most existing PLL works directly eliminate, rather than exploiting the label ambiguity, since they expl…

2022

Learngene: From Open-World to Your Learning Task

AAAI 2022technical

Although deep learning has made significant progress on fixed large-scale datasets, it typically encounters challenges regarding improperly detecting unknown/unseen classes in the open-world scenario, over-parametrized, and overfitting small samples. Since biological systems can overcome the above d…

2022

One Positive Label is Sufficient: Single-Positive Multi-Label Learning with Label Enhancement

NeurIPS 2022accept

Multi-label learning (MLL) learns from the examples each associated with multiple labels simultaneously, where the high cost of annotating all relevant labels for each training example is challenging for real-world applications. To cope with the challenge, we investigate single-positive multi-label…

2020

Label Distribution Learning on Auxiliary Label Space Graphs for Facial Expression Recognition

CVPR 2020poster

Many existing studies reveal that annotation inconsistency widely exists among a variety of facial expression recognition (FER) datasets. The reason might be the subjectivity of human annotators and the ambiguous nature of the expression labels. One promising strategy tackling such a problem is a re…

Cited by 244PDFScholar
2020

Progressive Identification of True Labels for Partial-Label Learning

ICML 2020poster

Partial-label learning (PLL) is a typical weakly supervised learning problem, where each training instance is equipped with a set of candidate labels among which only one is the true label. Most existing methods elaborately designed learning objectives as constrained optimizations that must be solve…

2019

Multi-Person Pose Estimation With Enhanced Channel-Wise and Spatial Information

CVPR 2019poster

Multi-person pose estimation is an important but challenging problem in computer vision. Although current approaches have achieved significant progress by fusing the multi-scale feature maps, they pay little attention to enhancing the channel-wise and spatial information of the feature maps. In this…

Cited by 189PDFScholar