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 language | English |
|---|---|
| Pages (from-to) | 3439-3450 |
| Journal | IEEE Transactions on Medical Imaging |
| Volume | 44 |
| Issue number | 8 |
| Online published | 1 May 2025 |
| DOIs | |
| Publication status | Published - 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
- APA
- Author
- BIBTEX
- Harvard
- Standard
- RIS
- Vancouver