Fourier Asymmetric Attention on Domain Generalization for Pan-Cancer Drug Response Prediction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Song, Ran, Bai, Yinpu, Liu, Hui
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916771150168064
author Song, Ran
Bai, Yinpu
Liu, Hui
author_facet Song, Ran
Bai, Yinpu
Liu, Hui
contents The accurate prediction of drug responses remains a formidable challenge, particularly at the single-cell level and in clinical treatment contexts. Some studies employ transfer learning techniques to predict drug responses in individual cells and patients, but they require access to target-domain data during training, which is often unavailable or only obtainable in future. In this study, we propose a novel domain generalization framework, termed FourierDrug, to address this challenge. Given the extracted feature from expression profile, we performed Fourier transforms and then introduced an asymmetric attention constraint that would cluster drug-sensitive samples into a compact group while drives resistant samples dispersed in the frequency domain. Our empirical experiments demonstrate that our model effectively learns task-relevant features from diverse source domains, and achieves accurate predictions of drug response for unseen cancer type. When evaluated on single-cell and patient-level drug response prediction tasks, FourierDrug--trained solely on in vitro cell line data without access to target-domain data--consistently outperforms or, at least, matched the performance of current state-of-the-art methods. These findings underscore the potential of our method for real-world clinical applications.
format Preprint
id arxiv_https___arxiv_org_abs_2502_04034
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Fourier Asymmetric Attention on Domain Generalization for Pan-Cancer Drug Response Prediction
Song, Ran
Bai, Yinpu
Liu, Hui
Machine Learning
Artificial Intelligence
The accurate prediction of drug responses remains a formidable challenge, particularly at the single-cell level and in clinical treatment contexts. Some studies employ transfer learning techniques to predict drug responses in individual cells and patients, but they require access to target-domain data during training, which is often unavailable or only obtainable in future. In this study, we propose a novel domain generalization framework, termed FourierDrug, to address this challenge. Given the extracted feature from expression profile, we performed Fourier transforms and then introduced an asymmetric attention constraint that would cluster drug-sensitive samples into a compact group while drives resistant samples dispersed in the frequency domain. Our empirical experiments demonstrate that our model effectively learns task-relevant features from diverse source domains, and achieves accurate predictions of drug response for unseen cancer type. When evaluated on single-cell and patient-level drug response prediction tasks, FourierDrug--trained solely on in vitro cell line data without access to target-domain data--consistently outperforms or, at least, matched the performance of current state-of-the-art methods. These findings underscore the potential of our method for real-world clinical applications.
title Fourier Asymmetric Attention on Domain Generalization for Pan-Cancer Drug Response Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2502.04034