Skip to content

Category

federated learning

367 papers

#machine learning Open access Nov 2025

A fused deep learning approach to transform drug repositioning

Drug repositioning holds promise for discovering new therapeutic applications for existing drugs, accelerating drug development and reducing associated costs. However, current methodologies encounter difficulties in managing diverse network representations, tackling cold start issues, and handling intrinsic attribute representations. Here we introduce a Unified Knowledge-Enhanced deep learning framework for Drug Repositioning (UKEDR), which integrates knowledge graph embedding, pre-training strategies, and recommendation systems to address these challenges. To overcome the cold start issue, UKEDR utilizes a semantic similarity-driven embedding approach. Our evaluations show that UKEDR performs better than various baselines, including classical machine learning, network-based, and deep learning approaches. In cold start scenarios, it demonstrates an improved capability in handling unseen nodes and generalizing to new compounds. The model also demonstrates strong robustness on imbalanced datasets and shows excellent generalization capabilities in specific drug-centric and disease-centric cold-start scenarios, validating its potential for real-world applications. Drug repositioning offers a promising avenue for accelerating drug development, yet existing methods struggle with network diversity, cold start issues, and intrinsic attribute representation. Here, the authors introduce UKEDR, a deep learning framework that integrates knowledge graph embedding and pre-training strategies to overcome the intractable cold start issue, achieving superior performance and interpretability in drug repurposing.

Kun Li, Jiacai Yi, Qing Ye et al. · 1 citation
#machine learning Review Open access Sep 2025

Enhanced Sampling in the Age of Machine Learning: Algorithms and Applications

Molecular dynamics simulations hold great promise for providing insight into the microscopic behavior of complex molecular systems. However, their effectiveness is often constrained by long timescales associated with rare events. Enhanced sampling methods have been developed to address these challenges, and recent years have seen a growing integration with machine learning techniques. This Review provides a comprehensive overview of how they are reshaping the field, with a particular focus on the data-driven construction of collective variables. Furthermore, these techniques have also improved biasing schemes and unlocked novel strategies via reinforcement learning and generative approaches. In addition to methodological advances, we highlight applications spanning different areas, such as biomolecular processes, ligand binding, catalytic reactions, and phase transitions. We conclude by outlining future directions aimed at enabling more automated strategies for rare-event sampling.

Kai Zhu, Enrico Trizio, Jintu Zhang et al. · 54 citations
#machine learning Open access Jun 2025

HiCLR: Knowledge-Induced Hierarchical Contrastive Learning with Retrosynthesis Prediction Yields a Reaction Foundation Model

Reaction representation learning is of paramount importance for adopting deep-learning-based chemistry modeling to solve real-world tasks such as synthesis planning. Most prevailing models are prestrained by self-supervised objectives that rely solely on the chemical structure information. Since structurally similar reactions could possess entirely distinct properties (e.g., reaction yields) and the synthesis-related tasks are highly heterogeneous, there are inherent limitations in constructing a foundational reaction model within the existing approaches. To tackle this limitation, we propose HiCLR, a knowledge-induced hierarchical contrastive learning framework for chemical reactions, by introducing relational inductive bias to forge chemically meaningful and generally applicable reaction fingerprints. Critically, the pretraining scheme combining both retrosynthesis prediction and contrastive loss enables HiCLR to tackle generation-based and understanding-based tasks simultaneously. Comprehensive experiments demonstrate that HiCLR successfully organizes the reaction space into hierarchical global semantic clusters, aligned well with prior knowledge. Consequently, HiCLR is the first foundation model that can be broadly applied to various synthesis-related tasks, and it achieves state-of-the-art performance in reaction classification, reaction condition recommendation, reaction yield prediction, synthesis planning, and even molecular property prediction. HiCLR demonstrates clear benefits in incorporating domain knowledge to guide the learning of neural networks, expediting AI-driven advancements in chemistry.

Jialu Wu, Yiheng Zhu, Xiaorui Wang et al. · 0 citations
#machine learning Open access Jul 2025

RSGPT: a generative transformer model for retrosynthesis planning pre-trained on ten billion datapoints

Retrosynthesis planning is a crucial task in organic synthesis, and deep-learning methods have enhanced and accelerated this process. With the advancement of the emergence of large language models, the demand for data is rapidly increasing. However, available retrosynthesis data are limited to only millions. Therefore, we pioneer the utilization of the template-based algorithm to generate chemical reaction data, resulting in the production of over 10 billion reaction datapoints. A generative pretrained transformer model is subsequently developed for template-free retrosynthesis planning by pre-training on 10 billion generated data. Inspired by the strategies of large language models, we introduce reinforcement learning to capture the relationships among products, reactants, and templates more accurately. Experiments demonstrate that our model achieves state-of-the-art performance on the benchmark, with a Top-1 accuracy of 63.4%, substantially outperforming previous models. Computer-aided synthesis-planning methods have significantly assisted synthesis planning. In this work, the authors present RSGPT, a generative model pre-trained on ten billion data points, achieving state-of-the-art performance for synthesis planning

Yafeng Deng, Xinda Zhao, Hanyu Sun et al. · 17 citations · ⚡2
#machine learning Open access Jun 2025

AntiBMPNN: Structure‐Guided Graph Neural Networks for Precision Antibody Engineering

Antibodies are crucial for medical applications, yet traditional methods for designing sequences are inefficient. This study introduces AntiBMPNN, an advanced deep‐learning framework that leverages an antibody‐specific 3D dataset, a fine‐tuned message‐passing neural network (MPNN), a frequency‐based scoring function, and AlphaFold 3 to achieve highly accurate antibody sequence design. AntiBMPNN surpasses ProteinMPNN with a perplexity of 1.5 and over 80% sequence recovery. Its scoring function, combined with AlphaFold 3, effectively prioritizes sequences based on structural recovery, positional stability, and biochemical or complex properties. Experimental validation highlights a 75% success rate in single‐point antibody design. AntiBMPNN consistently outperforms AbMPNN, AntiFold, and ProteinMPNN in designing complementarity determining regions (CDR) 1‐3, yielding stronger binding affinities. For CDR1 of huJ3 (anti‐HIV nanobody), it achieves a half maximal effective concentration (EC₅₀) of 9.2 nM (nanomolar), better than ProteinMPNN (135.2 nM) and AntiFold (59.3 nM), and comparable to AbMPNN (6.6 nM). For CDR2 of the D6 nanobody (targeting CD16), AntiBMPNN reaches 0.3 nM, outperforming AbMPNN (2.3 nM), AntiFold (0.7 nM), and ProteinMPNN (0.7 nM). In CDR3 of huJ3, it achieves 1.7 nM, surpassing AbMPNN (51.2 nM), with no detectable activity from AntiFold or ProteinMPNN. These findings confirm that AntiBMPNN‐designed sequences for J3 and D6 outperform the originals, highlighting its potential to improve therapeutic antibody design.

Ze-Yu Sun, Jiayi Yuan, Divya Jaiswal et al. · 9 citations
#machine learning Open access Aug 2026

AI-driven PROTAC design overcomes oncogenic resilience by eliminating the CLIP1-LTK fusion protein.

The discovery of CAP-Gly domain-containing linker protein 1(CLIP1)-Leukocyte tyrosine kinase (LTK) as an oncogenic fusion reveals a unique dependency not only on LTK kinase activity but also on CLIP1-mediated multimerization, a noncatalytic function that drives oncogenic signaling. While this fusion is currently targeted with anaplastic lymphoma kinase inhibitors, their exclusive focus on kinase inhibition leaves the scaffolding function intact, necessitating a complete protein clearance strategy. Here, we report the AI-guided development of a first-in-class proteolysis-targeting chimera (PROTAC) designed to selectively degrade the CLIP1-LTK fusion protein. By integrating deep learning models for ternary complex prediction with structure-based molecular optimization, we designed DCL05, an orally bioavailable degrader of CLIP1-LTK fusion protein, achieving picomolar degradation potency (DC50 = 40 pM) and robust antitumor activity. DCL05 consistently outperformed existing kinase inhibitors across a broad spectrum of LTK resistance-associated mutations, both in vitro and in vivo. Collectively, our study explores resistance-associated contexts of LTK and establishes a structure-guided PROTAC development pipeline, providing a promising therapeutic strategy for overcoming acquired resistance in kinase-driven cancers.

Shicheng Chen, Haiting Duan, S. Zhong et al. · 0 citations
#machine learning Open access Apr 2026

Accurate and task-agnostic modeling of enzymatic reactions through multimodal relational learning

Enzymatic reactions play an emerging role in a broad spectrum of scientific and industrial applications. The inherent complexity of enzymes, such as their substrate specificity, conformational flexibility, and the vast diversity of reactions involved, poses substantial challenges for the advanced computational prediction of enzymatic reactions with desirable accuracy. Moreover, existing approaches are mostly tailored for a specific sub-task, such as substrate prediction or binding site annotation, which limits their applicability. In this study, we introduce ERAM, a task-agnostic multimodal learning framework capable of addressing a broad range of downstream applications with both accuracy and efficiency. ERAM aligns pre-trained molecular representations from Protein Language Model with the knowledge of enzyme catalysis by modeling enzymatic reactions as multi-relational data. In enzyme retrieval tasks, ERAM achieves an improvement of 28.31% in mean average precision compared with the state-of-the-art (SOTA) method, CREEP. In substrate prediction tasks, ERAM outperforms the SOTA method ESP, achieving average improvements of 35.53% and 22.97% in Matthews correlation coefficient across two datasets. Additionally, ERAM exhibits commendable interpretability by assigning higher attention weights to binding sites, resulting in lower false-positive rates (42.36%) and higher overlap scores (70.59%) in the unsupervised binding site prediction task compared to RXNAA Mapper. By learning embeddings of substrates, enzymes, and products within a unified knowledge graph latent space, ERAM demonstrates its potential as a versatile and effective tool for enzyme catalysis research.

Yuansheng Huang, Lanqing Li, Wenjia Qian et al. · 2 citations
#machine learning Open access Jul 2026

BBBP-Atlas: Unified Interpretable Modeling of Blood–Brain Barrier Permeability across Small Molecules and Peptides

Accurate prediction of blood-brain barrier permeability (BBBP) is essential for central nervous system drug discovery, yet existing models are often limited by their reliance on predefined physicochemical descriptors, small-molecule-centered training sets, or conformation-dependent representations, which restricts their transferability across chemically diverse modalities especially peptides. In addition, publicly available BBBP datasets remain fragmented, inconsistently standardized, and weakly controlled for molecular redundancy, increasing the risk of data leakage and overestimated model performance. In this study, we propose BBBP-Atlas, a structure-aware BBB permeability prediction model designed for unified modeling of small molecules and peptides with the first cross-modal dataset OmniBBBP. Designed to bypass descriptor and conformation dependencies, our model represents standardized molecular structures as atom-level graphs to capture local atom-bond environments and long-range topological dependencies associated with BBB transport. This design enables direct learning of structure-permeability relationships from molecular topology. For model training and evaluation, we curated a cross-modal, redundancy-filtered database OmniBBBP that seamlessly unifies small molecules and complex peptides, containing 10,218 unique compounds with 9,316 small molecules and 902 peptides. BBBP-Atlas achieved an accuracy of 0.8914 and an MCC of 0.7678 on the independent test set. On a balanced external benchmark of 200 compounds, our model reached an AUC of 0.9108, an accuracy of 0.8500, and an MCC of 0.7000, outperforming LightBBB by an absolute MCC gain of 6%. Case studies further showed that BBBP-Atlas captured clinically meaningful BBB permeability patterns, correctly identifying lorlatinib as BBB-permeable and vancomycin as BBB-impermeable with high confidence. The OmniBBBP-backed BBBP-Atlas offers a versatile and cross-modal approach for single-compound prediction, batch screening, and dataset exploration for CNS drug discovery. BBBP-Atlas is available at https://cadd.drugflow.com/bbbp/.

Xin Shen, Qun Su, Hao Luo et al. · 0 citations
#machine learning Open access Jun 2026

Targeting the intrinsically disordered AR-NTD through a machine learning-based enhanced sampling workflow

Targeting the intrinsically disordered N-terminal domain of the androgen receptor (AR-NTD) represents a promising strategy to overcome resistance in prostate cancer. However, its inherent lack of a stable tertiary structure and highly dynamic conformational ensemble pose formidable challenges for rational drug design. This study introduces an integrated computational workflow that combines enhanced sampling techniques and machine learning collective variables to identify druggable conformations of the AR-NTD and elucidate the binding mechanism of its modulator, EPI-002. We characterize nine metastable states of the Tau-5 region and reveal that ligand recognition is driven by π–π stacking and structured water-mediated hydrogen bonds. Leveraging these insights, we perform structure-based virtual screening based on the identified druggable conformations and identify K53, a rationally designed AR-NTD antagonist, which exhibits potent anti-proliferative activity in enzalutamide-resistant prostate cancer cells. K53 directly binds the AR-NTD, suppresses AR transcriptional activity, and demonstrates high selectivity for cancer cells. This work provides a rational design paradigm for targeting intrinsically disordered proteins and offers a therapeutic candidate for resistant prostate cancer. In this work, the authors develop a machine learning–based enhanced sampling workflow to target the intrinsically disordered AR-NTD, identifying druggable conformations and enabling transferable modeling of ligand binding for rational drug discovery.

Kai Zhu, Huating Wang, Jintu Zhang et al. · 0 citations
#federated learning Open access Aug 2026

Information-Theoretic Framework for Trustworthy Federated Learning

Federated Learning (FL) offers a promising paradigm for decentralized machine learning, enabling collaborative model training without direct data sharing. However, traditional privacy-preserving techniques within FL often rely on ad-hoc assumptions and lack a rigorous theoretical basis. This work introduces an information-theoretic framework to address this limitation. We define a "Privacy Loss Function" predicated on mutual information between local models and global updates, providing a quantifiable measure of information leakage. The framework leverages established techniques such as differential privacy and homomorphic encryption to minimize this loss, ultimately leading to more robust and trustworthy FL systems. Our approach moves beyond intuitive notions of privacy, offering a mathematically sound foundation for designing and analyzing FL protocols, facilitating the development of truly secure and efficient distributed learning solutions. The core contribution is the formalization of privacy risk in FL using information-theoretic principles, enabling a more precise understanding and control over data leakage.

Jincheng Zhang · 0 citations
#federated learning Open access Aug 2026

Privacy-Enhancing Federated Learning Models for Cybersecurity in IoT Networks

The rapid expansion of the Internet of Things (IoT) has intensified cybersecurity risks by exposing distributed connected devices to increasingly complex and pervasive threats. Conventional centralized security mechanisms often struggle to accommodate the heterogeneous and decentralized structure of IoT networks. This study investigates Federated Learning (FL) as a decentralized approach to intrusion detection that enables local model training on IoT edge devices while transmitting only encrypted model updates to a central server, thereby preserving data privacy and reducing communication overhead. A novel FL-based Intrusion Detection System (IDS) architecture was developed using Convolutional Neural Networks (CNNs) for anomaly detection and the Federated Averaging (FedAvg) algorithm for aggregating local model updates. The framework was evaluated on standard IoT datasets under non-independent and identically distributed (non-IID) data conditions to simulate heterogeneous real-world environments. Experimental results demonstrate that the proposed system achieved a detection accuracy of 94.6%, an F1-score of 93.8%, and a recall of 92.7%, outperforming centralized and standalone local learning methods. The framework also reduced communication overhead by 35% and achieved convergence 28% faster than conventional approaches. These findings demonstrate that FL can provide a scalable, privacy-preserving, and computationally efficient foundation for strengthening IoT cybersecurity. This study contributes a decentralized machine-learning architecture for real-time, adaptive, and privacy-conscious intrusion detection in large-scale IoT environments.

Mohammed Ajuji, Yusuf Musa Malgwi, Asabe Sandra Ahmadu et al. · 0 citations

From tech blogs

See all →
MIT News · Artificial Intelligence Aug 27, 2026

Looking beyond natural sequences

A new machine-learning framework aims to improve the success rate of computational protein design while moving away from results that reproduce sequences found in nature.