Skip to main navigation Skip to search Skip to main content

LLM-guided Decoupled Probabilistic Prompt for Continual Learning in Medical Image Diagnosis

  • Yiwen Luo
  • , Wuyang Li
  • , Cheng Chen
  • , Xiang Li
  • , Tianming Liu
  • , Tianye Niu
  • , Yixuan Yuan*
  • *Corresponding author for this work

Research output: Journal Publications and Reviews › RGC 21 - Publication in refereed journal › peer-review

Abstract

Deep learning-based traditional diagnostic models typically exhibit limitations when applied to dynamic clinical environments that require handling the emergence of new diseases. Continual learning (CL) offers a promising solution, aiming to learn new knowledge while preserving previously learned knowledge. Though recent rehearsal-free CL methods employing prompt tuning (PT) have shown promise, they rely on deterministic prompts that struggle to handle diverse fine-grained knowledge. Moreover, existing PT methods utilize randomly initialized prompts that are trained under standard classification constraints, impeding expert knowledge integration and optimal performance acquisition. In this paper, we propose an LLM-guided Decoupled Probabilistic Prompt (LDPP) for Continual Learning in medical image diagnosis. Specifically, we develop an Expert Knowledge Generation (EKG) module that leverages LLM to acquire decoupled expert knowledge and comprehensive category descriptions. Then, we introduce a Decoupled Probabilistic Prompt pool (DePP) to construct a shared decoupled probabilistic prompt pool, which constructs a shared prompt pool with probabilistic prompts derived from the expert knowledge set. These prompts dynamically provide diverse and flexible descriptions for input images. Finally, We design a Steering Prompt Pool (SPP) to enhance intra-class compactness and promote model performance by learning nonshared prompts. With extensive experimental validation, LDPP consistently sets state-of-the-art performance under the challenging class-incremental setting in CL. © 2025 IEEE.
Original languageEnglish
Pages (from-to)3439-3450
JournalIEEE Transactions on Medical Imaging
Volume44
Issue number8
Online published1 May 2025
DOIs
Publication statusPublished - Aug 2025

Funding

This work was supported by Innovation and Technology CommissionInnovation and Technology Fund ITS/229/22 and Hong Kong Research Grants Council (RGC) General Research Fund 14204321, 14220622.

Research Keywords

  • Continual learning
  • expert knowledge
  • prompt tuning

RGC Funding Information

  • RGC-funded

Fingerprint

Dive into the research topics of 'LLM-guided Decoupled Probabilistic Prompt for Continual Learning in Medical Image Diagnosis'. Together they form a unique fingerprint.

Cite this