Learning to Align Molecules and Proteins: A Geometry-Aware Approach to Binding Affinity

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Refahi, Mohammadsaleh, Sokhansanj, Bahrad A., Brown, James R., Rosen, Gail
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908558217445376
author Refahi, Mohammadsaleh
Sokhansanj, Bahrad A.
Brown, James R.
Rosen, Gail
author_facet Refahi, Mohammadsaleh
Sokhansanj, Bahrad A.
Brown, James R.
Rosen, Gail
contents Accurate prediction of drug-target binding affinity can accelerate drug discovery by prioritizing promising compounds before costly wet-lab screening. While deep learning has advanced this task, most models fuse ligand and protein representations via simple concatenation and lack explicit geometric regularization, resulting in poor generalization across chemical space and time. We introduce FIRM-DTI, a lightweight framework that conditions molecular embeddings on protein embeddings through a feature-wise linear modulation (FiLM) layer and enforces metric structure with a triplet loss. An RBF regression head operating on embedding distances yields smooth, interpretable affinity predictions. Despite its modest size, FIRM-DTI achieves state-of-the-art performance on the Therapeutics Data Commons DTI-DG benchmark, as demonstrated by an extensive ablation study and out-of-domain evaluation. Our results underscore the value of conditioning and metric learning for robust drug-target affinity prediction.
format Preprint
id arxiv_https___arxiv_org_abs_2509_20693
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning to Align Molecules and Proteins: A Geometry-Aware Approach to Binding Affinity
Refahi, Mohammadsaleh
Sokhansanj, Bahrad A.
Brown, James R.
Rosen, Gail
Machine Learning
Artificial Intelligence
Molecular Networks
Accurate prediction of drug-target binding affinity can accelerate drug discovery by prioritizing promising compounds before costly wet-lab screening. While deep learning has advanced this task, most models fuse ligand and protein representations via simple concatenation and lack explicit geometric regularization, resulting in poor generalization across chemical space and time. We introduce FIRM-DTI, a lightweight framework that conditions molecular embeddings on protein embeddings through a feature-wise linear modulation (FiLM) layer and enforces metric structure with a triplet loss. An RBF regression head operating on embedding distances yields smooth, interpretable affinity predictions. Despite its modest size, FIRM-DTI achieves state-of-the-art performance on the Therapeutics Data Commons DTI-DG benchmark, as demonstrated by an extensive ablation study and out-of-domain evaluation. Our results underscore the value of conditioning and metric learning for robust drug-target affinity prediction.
title Learning to Align Molecules and Proteins: A Geometry-Aware Approach to Binding Affinity
topic Machine Learning
Artificial Intelligence
Molecular Networks
url https://arxiv.org/abs/2509.20693