Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Zheqi Lv

dblp:326/8807 · DBLP profile ↗
← Back
27ranked-venue papers
6as first author
27since 2021 · last 2026
0000-0001-6529-8088ORCID · verified

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 16 · 2 first-author · 16 since 2021Graphics, computer vision, multimedia, augmented reality and games · 11 · 3 first-author · 11 since 2021Databases, data management, data science and information retrieval · 10 · 3 first-author · 10 since 2021Applied, interdisciplinary, general and emerging computing · 7 · 2 first-author · 7 since 2021Computer networks · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
20 papers
Efficient and distributed learning · 27% Transfer learning and domain adaptation · 14% Language models and text generation · 12%
Databases, data mining, and information retrieval
10 papers
Recommender systems · 91% Information retrieval · 9%
Interdisciplinary, comprehensive, and emerging computing
1 paper
Computing education · 100%

Topics — the 30 heaviest of 59, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Recommender systems
sequential recommendation
3.142025
Device-Cloud Collaborative Correction for On-Device Recommendation · IJCAI 2025
Intelligent Model Update Strategy for Sequential Recommendation · WWW 2024
Semantic Codebook Learning for Dynamic Recommendation Models · ACM Multimedia 2024
Recommender systems
large language model-based recommendation
1.922026
ThinkRec: Thinking-based recommendation via LLM · WWW 2026
Collaboration of Large Language Models and Small Recommendation Models for Device-Cloud Recommendation · KDD (1) 2025
Recommender systems
on-device recommendation
1.722025
Forward Once for All: Structural Parameterized Adaptation for Efficient Cloud-coordinated On-device Recommendation · KDD (1) 2025
Device-Cloud Collaborative Correction for On-Device Recommendation · IJCAI 2025
Machine learning › Transfer learning and domain adaptation
domain shift
1.622025
Tackling Device Data Distribution Real-time Shift via Prototype-based Parameter Editing · ACM Multimedia 2025
Revisiting the Domain Shift and Sample Uncertainty in Multi-source Active Domain Transfer · CVPR 2024
Machine learning › Efficient and distributed learning › distributed training › edge training
device-cloud collaborative learning
1.522025
Device-Cloud Collaborative Correction for On-Device Recommendation · IJCAI 2025
DUET: A Tuning-Free Device-Cloud Collaborative Parameters Generation Framework for Efficient Device Model Generalization · WWW 2023
Machine learning › Trustworthy machine learning
out-of-distribution generalization
1.522024
AuG-KD: Anchor-Based Mixup Generation for Out-of-Domain Knowledge Distillation · ICLR 2024
Learning to Reweight for Generalizable Graph Neural Network · AAAI 2024
Machine learning › Efficient and distributed learning
distributed training
1.422024
Intelligent Model Update Strategy for Sequential Recommendation · WWW 2024
DUET: A Tuning-Free Device-Cloud Collaborative Parameters Generation Framework for Efficient Device Model Generalization · WWW 2023
Machine learning › Transfer learning and domain adaptation
domain adaptation
1.122026
AuG-KD: Anchor-Based Mixup Generation for Out-of-Domain Knowledge Distillation · ICLR 2024
Mitigating Structural Knowledge Collapse in Domain-Specific LLMs via Morpheme-Aware KV-Aggregation · ACL (1) 2026
Machine learning › Deep learning architectures and training
attention mechanism
1.012026
Mitigating Structural Knowledge Collapse in Domain-Specific LLMs via Morpheme-Aware KV-Aggregation · ACL (1) 2026
Natural language and speech › Language models and text generation
chain-of-thought reasoning
1.012026
ThinkRec: Thinking-based recommendation via LLM · WWW 2026
Natural language and speech › Language models and text generation
large language model training
1.012026
Scaling and Transferability of Annealing Strategies in Large Language Model Training · AAAI 2026
Machine learning › Optimization for machine learning
learning rate schedule
1.012026
Scaling and Transferability of Annealing Strategies in Large Language Model Training · AAAI 2026
Machine learning › Reinforcement learning
policy optimization
1.012026
A Rolling Stone Gathers No Moss: Adaptive Policy Optimization for Stable Self-Evaluation in Large Multimodal Models · AAAI 2026
Machine learning › Reinforcement learning › reward design
reward hacking mitigation
1.012026
A Rolling Stone Gathers No Moss: Adaptive Policy Optimization for Stable Self-Evaluation in Large Multimodal Models · AAAI 2026
Natural language and speech › Language models and text generation › large language model evaluation › automatic evaluation
self-evaluation
1.012026
A Rolling Stone Gathers No Moss: Adaptive Policy Optimization for Stable Self-Evaluation in Large Multimodal Models · AAAI 2026
Natural language and speech › Language models and text generation
tokenization
1.012026
Mitigating Structural Knowledge Collapse in Domain-Specific LLMs via Morpheme-Aware KV-Aggregation · ACL (1) 2026
Machine learning › Optimization for machine learning › learning rate schedule
warmup-stable-decay schedule
1.012026
Scaling and Transferability of Annealing Strategies in Large Language Model Training · AAAI 2026
Recommender systems
generative recommendation
1.012026
RASTP: Representation-Aware Semantic Token Pruning for Generative Recommendation with Semantic Identifiers · WWW 2026
Recommender systems › generative recommendation
semantic ID
1.012026
RASTP: Representation-Aware Semantic Token Pruning for Generative Recommendation with Semantic Identifiers · WWW 2026
Information retrieval
token pruning
1.012026
RASTP: Representation-Aware Semantic Token Pruning for Generative Recommendation with Semantic Identifiers · WWW 2026
Machine learning › Generative modeling
diffusion model
0.912025
Modality-Aware Diffusion Augmentation with Consistent Subspace Disentanglement for Session-based Recommendation · KDD (2) 2025
Machine learning › Trustworthy machine learning
fairness
0.912025
Disentangled Knowledge Tracing for Alleviating Cognitive Bias · WWW 2025
Knowledge, reasoning and agents › Knowledge representation and reasoning › knowledge engineering
knowledge integration
0.912025
Optimize Incompatible Parameters Through Compatibility-aware Knowledge Integration · AAAI 2025
Machine learning › Time series and sequential data › temporal sequence modeling
knowledge tracing
0.912025
Disentangled Knowledge Tracing for Alleviating Cognitive Bias · WWW 2025
Machine learning › Efficient and distributed learning
model merging
0.912025
Optimize Incompatible Parameters Through Compatibility-aware Knowledge Integration · AAAI 2025
Machine learning › Efficient and distributed learning
on-device inference
0.912025
Forward Once for All: Structural Parameterized Adaptation for Efficient Cloud-coordinated On-device Recommendation · KDD (1) 2025
Machine learning › Efficient and distributed learning › edge computing › on-device machine learning
on-device learning
0.912025
Tackling Device Data Distribution Real-time Shift via Prototype-based Parameter Editing · ACM Multimedia 2025
Machine learning › Transfer learning and domain adaptation
test-time adaptation
0.912025
Tackling Device Data Distribution Real-time Shift via Prototype-based Parameter Editing · ACM Multimedia 2025
Computing education
intelligent tutoring systems
0.912025
Cuff-KT: Tackling Learners' Real-time Learning Pattern Adjustment via Tuning-Free Knowledge State Guided Model Updating · KDD (2) 2025
Computing education › student modeling
knowledge tracing
0.912025
Cuff-KT: Tackling Learners' Real-time Learning Pattern Adjustment via Tuning-Free Knowledge State Guided Model Updating · KDD (2) 2025

Methods — techniques the papers use, named apart from their topics

reasoning trace injection · 2.0expert fusion · 2.0vocabulary expansion · 1.0semantic token pruning · 1.0representation-aware pruning · 1.0reinforcement learning · 1.0mixture of experts · 1.0learning rate annealing · 1.0adaptive reward model · 1.0adapter · 1.0KL regularization · 1.0tuning-free model updating · 0.9test-time training layers · 0.9self-correction network · 0.9parameter compatibility assessment · 0.9one forward pass adaptation · 0.9linear feature extraction · 0.9gradient conflict mitigation · 0.9
YearPublicationVenuePosition
2026 Scaling and Transferability of Annealing Strategies in Large Language Model Training
abstract
Learning rate scheduling is crucial for training large language models, yet understanding the optimal annealing strategies across different model configurations remains challenging. In this work, we investigate the transferability of annealing dynamics in large language model training and refine a generalized predictive framework for optimizing annealing strategies under the Warmup-Steady-Decay (WSD) scheduler. Our improved framework incorporates training steps, maximum learning rate, and annealing behavior, enabling more efficient optimization of learning rate schedules. Our work provides a practical guidance for selecting optimal annealing strategies without exhaustive hyperparameter searches, demonstrating that smaller models can serve as reliable proxies for optimizing the training dynamics of larger models. We validate our findings on extensive experiments using both Dense and Mixture-of-Experts (MoE) models, demonstrating that optimal annealing ratios follow consistent patterns and can be transferred across different training configurations.
Zhengyu Chen 0001, Teng Xiao, Zheqi Lv, Jinluan Yang, Jingang Wang
AAAI4
2026 A Rolling Stone Gathers No Moss: Adaptive Policy Optimization for Stable Self-Evaluation in Large Multimodal Models
abstract
Self-evaluation, a model's ability to assess the correctness of its own output, is crucial for Large Multimodal Models (LMMs) to achieve self-improvement in multi-turn conversations, yet largely absent in foundation models. Recent work has employed reinforcement learning (RL) to enhance self-evaluation; however, its fixed reward mechanism suffers from reward hacking when optimizing multiple training objectives, leading to model collapse. In this paper we propose AdaPO, an online reinforcement learning framework capable of adaptively adjusting training objective in real time according to the current training state for each task. Specifically, to mitigate reward hacking , AdaPO introduces an Adaptive Reward Model (ARM) and a Reward Aware Dynamic KL Regularization mechanism. ARM assesses the task's training state from the distribution of model generated multi-turn trajectories' performance. Reward Aware Dynamic KL replaces a fixed penalty with dynamic coefficients which is modulated by the reward gap between different multi-turn situations. Notably, our method automatically and smoothly adjusts its learning focus based on sub-tasks' training progress without manual intervention. Extensive experiments over 8 benchmarks and various models show that our method significantly enhances both direct reasoning and self-evaluation capability.
Hongcan Guo, Zheqi Lv, Shengyu Zhang 0001
AAAI3
2026 Mitigating Structural Knowledge Collapse in Domain-Specific LLMs via Morpheme-Aware KV-Aggregation
abstract
Standard tokenizers over-fragment domain terms, disrupting morpheme semantics.We characterize this representational misalignment as Structural Knowledge Collapse (SKC), where attention mechanisms fail to reconstruct coherent concepts from fragmented inputs.While existing input-centric solutions like vocabulary expansion address this, they necessitate expensive embedding retraining and neglect internal attention compositionality.To this end, we introduce Morphemeaware KV-aggregation Attention (MorphKA), a lightweight adapter that dynamically consolidates fragments without tokenizer changes.Bypassing tokenizer retraining, MorphKA employs a dual-phase strategy-Input-Level Morpheme Aggregation (IMA) and Context-Aware KV-Aggregation (AMRF)-to stabilize morpheme spans and synthesize higher-order concepts.Experiments on medical and legal benchmarks show MorphKA outperforms vocabulary adaptation baselines by 3.2-4.6%,reaching 7.9% on high-fragmentation terms.Moreover, MorphKA reduces catastrophic interference on general capabilities by 18-22% with ∼80% fewer parameters than embedding retraining approaches.
Yuxuan Si, Zheqi Lv, Chengxi Zang, Zhengyu Chen 0001, Fei Wu 0001
ACL (1)2
2026 ThinkRec: Thinking-based recommendation via LLM
abstract
Recent advances in large language models (LLMs) have enabled more semantic-aware recommendations through natural language generation. Existing LLM for recommendation (LLM4Rec) methods mostly operate in a System 1-like manner, relying on superficial features to match similar items based on click history, rather than reasoning through deeper behavioral logic. This often leads to superficial and erroneous recommendations. Motivated by this, we propose ThinkRec, a thinking-based framework that shifts LLM4Rec from System 1 to System 2 (rational system). Technically, ThinkRec introduces a thinking activation mechanism that augments item metadata with keyword summarization and injects synthetic reasoning traces, guiding the model to form interpretable reasoning chains that consist of analyzing interaction histories, identifying user preferences, and making decisions based on target items. On top of this, we propose an instance-wise expert fusion mechanism to reduce the reasoning difficulty. By dynamically assigning weights to expert models based on users' latent features, ThinkRec adapts its reasoning path to individual users, thereby enhancing precision and personalization. Extensive experiments on real-world datasets demonstrate that ThinkRec significantly improves the accuracy and interpretability of recommendations. Our implementations are available at https://github.com/Yu-Qi-hang/ThinkRec.
Qihang Yu, Kairui Fu, Zheqi Lv, Shengyu Zhang 0001, Xinhui Wu, Chen Lin 0001, Bo Zheng 0007, Fei Wu 0001
WWW3
2026 RASTP: Representation-Aware Semantic Token Pruning for Generative Recommendation with Semantic Identifiers
Tianyu Zhan, Kairui Fu, Zheqi Lv, Shengyu Zhang 0001
WWW3
2026 MAKIMA: Tuning-free multi-attribute open-domain video editing via mask-guided attention modulation
Wenqiao Zhang, Hongyang He, Juncheng Li 0006, Zheqi Lv, Siliang Tang, Yueting Zhuang
Expert Syst. Appl.6
2025 Optimize Incompatible Parameters Through Compatibility-aware Knowledge Integration
abstract
Deep neural networks have become foundational to advancements in multiple domains, including recommendation systems, natural language processing, and so on. Despite their successes, these models often contain incompatible parameters that can be underutilized or detrimental to model performance, particularly when faced with specific, varying data distributions. Existing research excels in removing such parameters or merging the outputs of multiple different pretrained models. However, the former focuses on efficiency rather than performance, while the latter requires several times more computing and storage resources to support inference. In this paper, we set the goal to explicitly improve these incompatible parameters by leveraging the complementary strengths of different models, thereby directly enhancing the models without any additional parameters. Specifically, we propose Compatibility-aware Knowledge Integration (CKI), which consists of Parameter Compatibility Assessment and Parameter Splicing, which are used to evaluate the knowledge content of multiple models and integrate the knowledge into one model, respectively. The integrated model can be used directly for inference or for further fine-tuning. Extensive experiments on various recommendation and language datasets show that CKI can effectively optimize incompatible parameters under multiple tasks and settings to break through the training limit of the original model without increasing the inference cost.
Zheqi Lv, Keming Ye, Zishu Wei, Qi Tian 0003, Shengyu Zhang 0001, Wenqiao Zhang, Wenjie Wang 0007, Kun Kuang 0001, Tat-Seng Chua, Fei Wu 0001
AAAI1
2025 Preliminary Evaluation of the Test-Time Training Layers in Recommendation System (Student Abstract)
abstract
This paper explores the application and effectiveness of TestTime Training (TTT) layers in improving the performance of recommendation systems. We developed a model, TTT4Rec, utilizing TTT-Linear as the feature extraction layer. Our tests across multiple datasets indicate that TTT4Rec, as a base model, performs comparably or even surpasses other baseline models in similar environments.
Tianyu Zhan, Zheqi Lv, Shengyu Zhang 0001, Jiwei Li 0001
AAAI2
2025 Device-Cloud Collaborative Correction for On-Device Recommendation
abstract
With the rapid development of recommendation models and device computing power, device-based recommendation has become an important research area due to its better real-time performance and privacy protection. Previously, Transformer-based sequential recommendation models have been widely applied in this field because they outperform Recurrent Neural Network (RNN)-based recommendation models in terms of performance. However, as the length of interaction sequences increases, Transformer-based models introduce significantly more space and computational overhead compared to RNN-based models, posing challenges for device-based recommendation. To balance real-time performance and high performance on devices, we propose Device-Cloud Collaborative Correction Framework for On-Device Recommendation (CoCorrRec). CoCorrRec uses a self-correction network (SCN) to correct parameters with extremely low time cost. By updating model parameters during testing based on the input token, it achieves performance comparable to current optimal but more complex Transformer-based models. Furthermore, to prevent SCN from overfitting, we design a global correction network (GCN) that processes hidden states uploaded from devices and provides a global correction solution. Extensive experiments on multiple datasets show that CoCorrRec outperforms existing Transformer-based and RNN-based device recommendation models in terms of performance, with fewer parameters and lower FLOPs, thereby achieving a balance between real-time performance and high efficiency. Code is available at https: //github.com/Yuzt-zju/CoCorrRec.
Tianyu Zhan, Shengyu Zhang 0001, Zheqi Lv, Jieming Zhu, Jiwei Li 0001, Fan Wu 0006, Fei Wu 0001
IJCAI3
2025 Forward Once for All: Structural Parameterized Adaptation for Efficient Cloud-coordinated On-device Recommendation
abstract
In cloud-centric recommender system, regular data exchanges between user devices and cloud could potentially elevate bandwidth demands and privacy risks. On-device recommendation emerges as a viable solution by performing reranking locally to alleviate these concerns. Existing methods primarily focus on developing local adaptive parameters, while potentially neglecting the critical role of tailor-made model architecture. Insights from broader research domains suggest that varying data distributions might favor distinct architectures for better fitting. In addition, imposing a uniform model structure across heterogeneous devices may result in risking inefficacy on less capable devices or sub-optimal performance on those with sufficient capabilities. In response to these gaps, our paper introduces Forward-OFA, a novel approach for the dynamic construction of device-specific networks (both structure and parameters). Forward-OFA employs a structure controller to selectively determine whether each block needs to be assembled for each device. However, during the training of the structure controller, these assembled heterogeneous structures are jointly optimized, where the co-adaption among blocks might encounter gradient conflicts. To mitigate this, Forward-OFA is designed to establish a structure-guided mapping of real-time behaviors to individual parameters of assembled networks. Structure-related parameters and parallel components within the mapper prevent each part from receiving heterogeneous gradients from others, thus bypassing the gradient conflicts for coupled optimization. Besides, direct mapping enables Forward-OFA to achieve adaptation through only one forward pass, allowing for swift adaptation to changing interests and eliminating the requirement for on-device backpropagation. Further sophisticated design protects user privacy and makes the consumption of additional modules on device negligible. Experiments on real-world datasets demonstrate the effectiveness and efficiency of Forward-OFA.
Kairui Fu, Zheqi Lv, Shengyu Zhang 0001, Fan Wu 0006, Kun Kuang 0001
KDD (1)2
2025 Collaboration of Large Language Models and Small Recommendation Models for Device-Cloud Recommendation
abstract
Large Language Models (LLMs) for Recommendation (LLM4Rec) is a promising research direction that has demonstrated exceptional performance in this field. However, its inability to capture real-time user preferences greatly limits the practical application of LLM4Rec because (i) LLMs are costly to train and infer frequently, and (ii) LLMs struggle to access real-time data (its large number of parameters poses an obstacle to deployment on devices). Fortunately, small recommendation models (SRMs) can effectively supplement these shortcomings of LLM4Rec diagrams by consuming minimal resources for frequent training and inference, and by conveniently accessing real-time data on devices.
Zheqi Lv, Tianyu Zhan, Wenjie Wang 0007, Xinyu Lin 0001, Shengyu Zhang 0001, Wenqiao Zhang, Jiwei Li 0001, Kun Kuang 0001, Fei Wu 0001
KDD (1)1
2025 Modality-Aware Diffusion Augmentation with Consistent Subspace Disentanglement for Session-based Recommendation
abstract
Session-based Recommendation (SBR) explores dynamic user interests based on short anonymous sessions. The performance of traditional ID-collaborative SBR models is constrained by the limited session length. To this end, we focus on Multi-modal Session-based Recommendation (MMSBR), which aims to leverage modality knowledge to promote short-term user interest modeling. Previous studies cannot solve the MMSBR problem well due to issues as: (I1) Modal-invariant and specific dependencies are difficult to disentangle and unify. (I2) Multiple modal transitions produce mutual effects. (I3) Modal-aware personalized preference discrepancy exists. Thus, we propose a modality-aware diffusion-based framework MDSD with subspace disentanglement, which decouples multi-modal collaborations and provides a new paradigm that integrates modality representation generation into next-item prediction. We first employ modality-disentangled consistency graphs based on affinity semantics to extract the unified modal subspace. Then we design a cross-modal contrastive attention fusion that explores cross-modality interplay, enhancing modal alignment and consistency. To interpret personalized interest discrepancy, we realize the preference-guided multi-modal diffusion, which combines modality-aware interest generation with prediction based on collaborative signals, thus providing comprehensive recommendation results. Extensive experiments on real-world datasets demonstrate the effectiveness of MDSD.
Jiajie Su, Chaochao Chen 0001, Weiming Liu 0005, Yihao Wang 0007, Zheqi Lv, Jianwei Yin
KDD (2)6
2025 Cuff-KT: Tackling Learners' Real-time Learning Pattern Adjustment via Tuning-Free Knowledge State Guided Model Updating
abstract
Knowledge Tracing (KT) is a core component of Intelligent Tutoring Systems, modeling learners' knowledge state to predict future performance and provide personalized learning support. Traditional KT models assume that learners' learning abilities remain relatively stable over short periods or change in predictable ways based on prior performance. However, in reality, learners' abilities change irregularly due to factors like cognitive fatigue, motivation, and external stress--a task introduced, which we refer to as Real-time Learning Pattern Adjustment (RLPA). Existing KT models, when faced with RLPA, lack sufficient adaptability, because they fail to timely account for the dynamic nature of different learners' evolving learning patterns. Current strategies for enhancing adaptability rely on retraining, which leads to significant overfitting and high time overhead issues. To address this, we propose Cuff-KT, comprising a controller and a generator. The controller assigns value scores to learners, while the generator generates personalized parameters for selected learners. Cuff-KT controllably adapts to data changes fast and flexibly without fine-tuning. Experiments on five datasets from different subjects demonstrate that Cuff-KT significantly improves the performance of five KT models with different structures under intra- and inter-learner shifts, with an average relative increase in AUC of 10% and 4%, respectively, at a negligible time cost, effectively tackling RLPA task. Our code and datasets are fully available at https://github.com/zyy-2001/Cuff-KT.
Yiyun Zhou, Zheqi Lv, Shengyu Zhang 0001, Jingyuan Chen 0003
KDD (2)2
2025 Tackling Device Data Distribution Real-time Shift via Prototype-based Parameter Editing
abstract
The on-device real-time data distribution shift on devices challenges the generalization of lightweight on-device models. This critical issue is often overlooked in current research, which predominantly relies on data-intensive and computationally expensive fine-tuning approaches. To tackle this, we introduce Persona, a novel personalized method using a prototype-based, backpropagation-free parameter editing framework to enhance model generalization without post-deployment retraining. Persona employs a neural adapter in the cloud to generate a parameter editing matrix based on real-time device data. This matrix adeptly adapts on-device models to the prevailing data distributions, efficiently clustering them into prototype models. The prototypes are dynamically refined via the parameter editing matrix, facilitating efficient evolution. Furthermore, the integration of cross-layer knowledge transfer ensures consistent and context-aware multi-layer parameter changes and prototype assignment. Extensive experiments on vision task and recommendation task on multiple datasets confirm Persona's effectiveness and generality.
Zheqi Lv, Wenqiao Zhang, Kairui Fu, Qi Tian 0003, Shengyu Zhang 0001, Jiajie Su, Jingyuan Chen 0003, Kun Kuang 0001, Fei Wu 0001
ACM Multimedia1
2025 Disentangled Knowledge Tracing for Alleviating Cognitive Bias
abstract
In the realm of Intelligent Tutoring System (ITS), the accurate assessment of students' knowledge states through Knowledge Tracing (KT) is crucial for personalized learning. However, due to data bias, i.e., the unbalanced distribution of question groups ( e.g., concepts), conventional KT models are plagued by cognitive bias, which tends to result in cognitive underload for overperformers and cognitive overload for underperformers. More seriously, this bias is amplified with the exercise recommendations by ITS. After delving into the causal relations in the KT models, we identify the main cause as the confounder effect of students' historical correct rate distribution over question groups on the student representation and prediction score. Towards this end, we propose a Disentangled Knowledge Tracing (DisKT) model, which separately models students' familiar and unfamiliar abilities based on causal effects and eliminates the impact of the confounder in student representation within the model. Additionally, to shield the contradictory psychology ( e.g., guessing and mistaking) in the students' biased data, DisKT introduces a contradiction attention mechanism. Furthermore, DisKT enhances the interpretability of the model predictions by integrating a variant of Item Response Theory. Experimental results on 11 benchmarks and 3 synthesized datasets with different bias strengths demonstrate that DisKT significantly alleviates cognitive bias and outperforms 16 baselines in evaluation accuracy.
Yiyun Zhou, Zheqi Lv, Shengyu Zhang 0001, Jingyuan Chen 0003
WWW2
2025 FedMcon: an adaptive aggregation method for federated learning via meta controller
abstract
Federated learning (FL) emerged as a novel machine learning setting that enables collaboratively training deep models on decentralized clients with privacy constraints. In the vanilla federated averaging algorithm (FedAvg), the global model is generated by the weighted linear combination of local models, and the weights are proportional to the local data sizes. This methodology, however, encounters challenges when facing heterogeneous and unknown client data distributions, often leading to discrepancies from the intended global objective. The linear combination-based aggregation often fails to address the varied dynamics presented by diverse scenarios, settings, and data distributions inherent in FL, resulting in hindered convergence and compromised generalization. In this paper, we present a new aggregation method, FedMcon, within a framework of meta-learning for FL. We introduce a learnable controller trained on a small proxy dataset and served as an aggregator to learn how to adaptively aggregate heterogeneous local models into a better global model toward the desired objective. The experimental results indicate that the proposed method is effective on extremely non-independent and identically distributed data and it can simultaneously reach 19 times communication speedup in a single FL setting.
Tao Shen 0002, Zexi Li 0001, Ziyu Zhao 0001, Didi Zhu, Zheqi Lv, Kun Kuang 0001, Shengyu Zhang 0001, Chao Wu 0001, Fei Wu 0001
Frontiers Inf. Technol. Electron. Eng.5
2025 Backpropagation-Free Multi-modal On-Device Model Adaptation via Cloud-Device Collaboration
abstract
In our increasingly interconnected world, where intelligent devices continually amass copious personalized multi-modal data, a pressing need arises to deliver high-quality, personalized device-aware services. However, this endeavor presents a multifaceted challenge to prevailing AI systems primarily rooted in the cloud. As these systems grapple with shifting data distributions between the cloud and devices, the traditional approach of fine-tuning-based adaptation (FTA) exists the following issues: the costly and time-consuming data annotation required by FTA and the looming risk of model overfitting. To surmount these challenges, we introduce a universal on-device Multi-modal Model Adaptation framework, revolutionizing on-device model adaptation by striking a balance between efficiency and effectiveness. The framework features the Fast Domain Adaptor (FDA) hosted in the cloud, providing tailored parameters for the lightweight multi-modal model on devices. To enhance adaptability across multi-modal tasks, the AnchorFrame Distribution Reasoner (ADR) minimizes communication costs. Our contributions, encapsulated in the Cloud-Device Collaboration Multi-modal Parameter Generation (CDC-MMPG) framework, represent a pioneering solution for on-Device Multi-modal Model Adaptation (DMMA). Extensive experiments validate the efficiency and effectiveness of our method, particularly in video question answering and retrieval tasks, driving forward the integration of intelligent devices into our daily lives.
Wei Ji 0008, Li Li 0091, Zheqi Lv, Wenqiao Zhang, Mengze Li 0001, Wenqiang Lei, Roger Zimmermann
ACM Trans. Multim. Comput. Commun. Appl.3
2024 Learning to Reweight for Generalizable Graph Neural Network
abstract
Graph Neural Networks (GNNs) show promising results for graph tasks. However, existing GNNs' generalization ability will degrade when there exist distribution shifts between testing and training graph data. The fundamental reason for the severe degeneration is that most GNNs are designed based on the I.I.D hypothesis. In such a setting, GNNs tend to exploit subtle statistical correlations existing in the training set for predictions, even though it is a spurious correlation. In this paper, we study the problem of the generalization ability of GNNs on Out-Of-Distribution (OOD) settings. To solve this problem, we propose the Learning to Reweight for Generalizable Graph Neural Network (L2R-GNN) to enhance the generalization ability for achieving satisfactory performance on unseen testing graphs that have different distributions with training graphs. We propose a novel nonlinear graph decorrelation method, which can substantially improve the out-of-distribution generalization ability and compares favorably to previous methods in restraining the over-reduced sample size. The variables of graph representation are clustered based on the stability of their correlations, and graph decorrelation method learns weights to remove correlations between the variables of different clusters rather than any two variables. Besides, we introduce an effective stochastic algorithm based on bi-level optimization for the L2R-GNN framework, which enables simultaneously learning the optimal weights and GNN parameters, and avoids the over-fitting issue. Experiments show that L2R-GNN greatly outperforms baselines on various graph prediction benchmarks under distribution shifts.
Zhengyu Chen 0001, Teng Xiao, Kun Kuang 0001, Zheqi Lv, Min Zhang 0068, Jinluan Yang, Chengqiang Lu, Hongxia Yang, Fei Wu 0001
AAAI4
2024 Revisiting the Domain Shift and Sample Uncertainty in Multi-source Active Domain Transfer
abstract
Active Domain Adaptation (ADA) aims to maximally boost model adaptation in a new target domain by actively selecting a limited number of target data to annotate. This setting neglects the more practical scenario where training data are collected from multiple sources. This motivates us to extend ADA from a single source domain to multiple source domains, termed Multi-source Active Domain Adaptation (MADA). Not surprisingly, we find that most traditional ADA methods cannot work directly in such a setting, mainly due to the excessive domain gap introduced by all the source domains. Considering this, we propose a Detective framework that comprehensively considers the domain shift between multi-source domains and target domains to detect the informative target samples. Specifically, the Detective leverages a dynamic Domain Adaptation (DA) model that learns how to adapt the model's parameters to fit the union of multi-source domains. This enables an approximate single-source domain modeling by the dynamic model. We then compre-hensively measure both domain uncertainty and predictive uncertainty in the target domain to detect informative target samples using evidential deep learning, thereby mitigating uncertainty miscalibration. Experiments demonstrate that our solution outperforms existing methods by a considerable margin on three domain adaptation benchmarks. Our project is available at https://github.com/wannature/MADA.
Wenqiao Zhang, Zheqi Lv
CVPR2
2024 AuG-KD: Anchor-Based Mixup Generation for Out-of-Domain Knowledge Distillation
abstract
Due to privacy or patent concerns, a growing number of large models are released without granting access to their training data, making transferring their knowledge inefficient and problematic. In response, Data-Free Knowledge Distillation (DFKD) methods have emerged as direct solutions. However, simply adopting models derived from DFKD for real-world applications suffers significant performance degradation, due to the discrepancy between teachers' training data and real-world scenarios (student domain). The degradation stems from the portions of teachers' knowledge that are not applicable to the student domain. They are specific to the teacher domain and would undermine students' performance. Hence, selectively transferring teachers' appropriate knowledge becomes the primary challenge in DFKD. In this work, we propose a simple but effective method AuG-KD. It utilizes an uncertainty-guided and sample-specific anchor to align student-domain data with the teacher domain and leverages a generative method to progressively trade off the learning process between OOD knowledge distillation and domain-specific information learning via mixup learning. Extensive experiments in 3 datasets and 8 settings demonstrate the stability and superiority of our approach.
Zheqi Lv, Shengyu Zhang 0001, Xinyu Duan, Fei Wu 0001, Kun Kuang 0001
ICLR2
2024 DIET: Customized Slimming for Incompatible Networks in Sequential Recommendation
abstract
Due to the continuously improving capabilities of mobile edges, recommender systems start to deploy models on edges to alleviate network congestion caused by frequent mobile requests. Several studies have leveraged the proximity of edge-side to real-time data, fine-tuning them to create edge-specific models. Despite their significant progress, these methods require substantial on-edge computational resources and frequent network transfers to keep the model up to date. The former may disrupt other processes on the edge to acquire computational resources, while the latter consumes network bandwidth, leading to a decrease in user satisfaction. In response to these challenges, we propose a customizeD slImming framework for incompatiblE neTworks(DIET). DIET deploys the same generic backbone (potentially incompatible for a specific edge) to all devices. To minimize frequent bandwidth usage and storage consumption in personalization, DIET tailors specific subnets for each edge based on its past interactions, learning to generate slimming subnets(diets) within incompatible networks for efficient transfer. It also takes the inter-layer relationships into account, empirically reducing inference time while obtaining more suitable diets. We further explore the repeated modules within networks and propose a more storage-efficient framework, DIETING, which utilizes a single layer of parameters to represent the entire network, achieving comparably excellent performance. The experiments across four state-of-the-art datasets and two widely used models demonstrate the superior accuracy in recommendation and efficiency in transmission and storage of our framework.
Kairui Fu, Shengyu Zhang 0001, Zheqi Lv, Jingyuan Chen 0003, Jiwei Li 0001
KDD3
2024 De-fine: Decomposing and Refining Visual Programs with Auto-Feedback
abstract
Visual programming, a modular paradigm, integrates different modules and Python operators to solve various vision-language tasks. Unlike end-to-end models that need task-specific data, it performs visual processing and inference in an unsupervised manner. Current visual programming methods generate programs in a single pass where the ability to evaluate and optimize based on feedback, unfortunately, is lacking, which consequentially limits their effectiveness for complex, multi-step problems. Inspired by benders decomposition, we introduce De-fine, a training-free framework that automatically decomposes complex tasks into simpler subtasks and refines programs through auto-feedback. This model-agnostic approach can improve logical reasoning performance by integrating the strengths of multiple models. Our experiments across various visual tasks show that De-fine creates more robust programs. Moreover, viewing each feedback module as an independent agent will yield fresh prospects for the field of agent research.
Minghe Gao, Juncheng Li 0006, Hao Fei 0001, Liang Pang 0001, Wei Ji 0008, Guoming Wang, Zheqi Lv, Wenqiao Zhang, Siliang Tang, Yueting Zhuang
ACM Multimedia7
2024 Semantic Codebook Learning for Dynamic Recommendation Models
abstract
Dynamic sequential recommendation (DSR) can generate model parameters based on user behavior to improve the personalization of sequential recommendation under various user preferences. However, it faces the challenges of large parameter search space and sparse and noisy user-item interactions, which reduces the applicability of the generated model parameters. The Semantic Codebook Learning for Dynamic Recommendation Models (SOLID) framework presents a significant advancement in DSR by effectively tackling these challenges. By transforming item sequences into semantic sequences and employing a dual parameter model, SOLID compresses the parameter generation search space and leverages homogeneity within the recommendation system. The introduction of the semantic metacode and semantic codebook, which stores disentangled item representations, ensures robust and accurate parameter generation. Extensive experiments demonstrates that SOLID consistently outperforms existing DSR, delivering more accurate, stable, and robust recommendations.
Zheqi Lv, Shaoxuan He, Tianyu Zhan, Shengyu Zhang 0001, Wenqiao Zhang, Jingyuan Chen 0003, Zhou Zhao 0001, Fei Wu 0001
ACM Multimedia1
2024 Intelligent Model Update Strategy for Sequential Recommendation
abstract
Modern online platforms are increasingly employing recommendation systems to address information overload and improve user engagement. There is an evolving paradigm in this research field that recommendation network learning occurs both on the cloud and on edges with knowledge transfer in between (i.e., edge-cloud collaboration). Recent works push this filed further by enabling edge-specific context-aware adaptivity, where model parameters are updated in real-time based on incoming on-edge data. However, we argue that frequent data exchanges between the cloud and edges often lead to inefficiency and waste of communication/computation resources, as considerable parameter updates might be redundant. To investigate this problem, we introduce Intelligent Edge-Cloud Parameter Request Model (IntellectReq). IntellectReq is designed to operate on edge, evaluating the cost-benefit landscape of parameter requests with minimal computation and communication overhead. We formulate this as a novel learning task, aimed at the detection of out-of-distribution data, thereby fine-tuning adaptive communication strategies. Further, we employ statistical mapping techniques to convert real-time user behavior into a normal distribution, thereby employing multi-sample outputs to quantify the model's uncertainty and thus its generalization capabilities. Rigorous empirical validation on four widely-adopted benchmarks evaluates our approach, evidencing a marked improvement in the efficiency and generalizability of edge-cloud collaborative and dynamic recommendation systems.
Zheqi Lv, Wenqiao Zhang, Zhengyu Chen 0001, Shengyu Zhang 0001, Kun Kuang 0001
WWW1
2023 ART: rule bAsed futuRe-inference deducTion
abstract
Mengze Li, Tianqi Zhao, Bai Jionghao, Baoyi He, Jiaxu Miao, Wei Ji, Zheqi Lv, Zhou Zhao, Shengyu Zhang, Wenqiao Zhang, Fei Wu. Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing. 2023.
Mengze Li 0001, Jionghao Bai, Baoyi He, Jiaxu Miao, Wei Ji 0008, Zheqi Lv, Zhou Zhao 0001, Shengyu Zhang 0001, Wenqiao Zhang, Fei Wu 0001
EMNLP7
2023 DUET: A Tuning-Free Device-Cloud Collaborative Parameters Generation Framework for Efficient Device Model Generalization
abstract
Device Model Generalization (DMG) is a practical yet under-investigated research topic for on-device machine learning applications. It aims to improve the generalization ability of pre-trained models when deployed on resource-constrained devices, such as improving the performance of pre-trained cloud models on smart mobiles. While quite a lot of works have investigated the data distribution shift across clouds and devices, most of them focus on model fine-tuning on personalized data for individual devices to facilitate DMG. Despite their promising, these approaches require on-device re-training, which is practically infeasible due to the overfitting problem and high time delay when performing gradient calculation on real-time data. In this paper, we argue that the computational cost brought by fine-tuning can be rather unnecessary. We consequently present a novel perspective to improving DMG without increasing computational cost, i.e., device-specific parameter generation which directly maps data distribution to parameters. Specifically, we propose an efficient Device-cloUd collaborative parametErs generaTion framework (DUET). DUET is deployed on a powerful cloud server that only requires the low cost of forwarding propagation and low time delay of data transmission between the device and the cloud. By doing so, DUET can rehearse the device-specific model weight realizations conditioned on the personalized real-time data for an individual device. Importantly, our DUET elegantly connects the cloud and device as a “duet” collaboration, frees the DMG from fine-tuning, and enables a faster and more accurate DMG paradigm. We conduct an extensive experimental study of DUET on three public datasets, and the experimental results confirm our framework’s effectiveness and generalisability for different DMG tasks.
Zheqi Lv, Wenqiao Zhang, Shengyu Zhang 0001, Kun Kuang 0001, Feng Wang 0072, Zhengyu Chen 0001, Tao Shen 0002, Hongxia Yang, Beng Chin Ooi, Fei Wu 0001
WWW1
2023 Federated mutual learning: a collaborative machine learning method for heterogeneous data, models, and objectives
abstract
Federated learning (FL) is a novel technique in deep learning that enables clients to collaboratively train a shared model while retaining their decentralized data. However, researchers working on FL face several unique challenges, especially in the context of heterogeneity. Heterogeneity in data distributions, computational capabilities, and scenarios among clients necessitates the development of customized models and objectives in FL. Unfortunately, existing works such as FedAvg may not effectively accommodate the specific needs of each client. To address the challenges arising from heterogeneity in FL, we provide an overview of the heterogeneities in data, model, and objective (DMO). Furthermore, we propose a novel framework called federated mutual learning (FML), which enables each client to train a personalized model that accounts for the data heterogeneity (DH). A “meme model” serves as an intermediary between the personalized and global models to address model heterogeneity (MH). We introduce a knowledge distillation technique called deep mutual learning (DML) to transfer knowledge between these two models on local data. To overcome objective heterogeneity (OH), we design a shared global model that includes only certain parts, and the personalized model is task-specific and enhanced through mutual learning with the meme model. We evaluate the performance of FML in addressing DMO heterogeneities through experiments and compare it with other commonly used FL methods in similar scenarios. The results demonstrate that FML outperforms other methods and effectively addresses the DMO challenges encountered in the FL setting.
Tao Shen 0002, Jie Zhang 0081, Xinkang Jia, Fengda Zhang, Zheqi Lv, Kun Kuang 0001, Chao Wu 0001, Fei Wu 0001
Frontiers Inf. Technol. Electron. Eng.5