MTL-DFM: Multi-Task Learning and Diffusion Model for ISAC Systems

Deep learning (DL) has emerged as a key enabler for unlocking the potential of integrated sensing and communication (ISAC). Despite recent progress, current DL methods primarily handle sensing and communication as independent tasks, overlooking potential performance enhancement through a joint approach. Moreover, existing methods rely on fully annotated data for training, which is often challenging to obtain, especially in multi-task scenarios where labeled data may be scarce or only exist for a subset of tasks. Motivated by these shortcomings, this paper proposes a novel scheme, MTL-DFM, to enable simultaneous sensing and communication with partially labeled training data, which leverages multi-task learning (MTL) and a diffusion model (DFM). In particular, we introduce an initial feature extraction module (IFEM) to jointly capture shared information across tasks and explore inherent cross-task connections for enhanced feature extraction. Next, we design a signal denoising with incomplete labeling (SDIL) module to effectively remove noise from extracted information and construct comprehensive feature representations for all tasks with partially labeled datasets, which is difficult for conventional DL methods. Simulation results verify the superior performance offered by MTL-DFM over prior state-of-the-art methods.