Search papers, labs, and topics across Lattice.
This paper introduces PETA, a parameter-efficient framework for test-time adaptation of pretrained virtual screening models, specifically targeting the ranking of active ligands for protein pockets. By leveraging molecular diffusion and chemical validity filtering to create pocket-specific negatives, PETA enhances the ranking process without the need for extensive retraining, thus minimizing computational costs. Experimental results show that PETA significantly outperforms both pretrained and fully retrained models while only updating a minimal fraction of the model's parameters, demonstrating its effectiveness in improving ligand ranking accuracy.
PETA achieves superior ligand ranking by adapting pretrained models at test time with only 0.03% of the parameters updated, revolutionizing efficiency in virtual screening.
Accurately ranking active ligands for a target protein pocket from massive chemical libraries remains a central challenge in virtual screening. DrugCLIP and its recent extensions substantially accelerate this process by encoding protein pockets and molecules into a shared embedding space. Despite this progress, further performance improvements typically require retraining the entire model, incurring substantial computational overhead and making target-specific customization inefficient. In this work, we formulate the specialization of pretrained virtual screening models to individual pockets as a test-time adaptation problem and propose PETA, a parameter-efficient framework that directly adapts pretrained model at test time. Given a target pocket, PETA constructs pocket-specific negatives through molecular diffusion and chemical validity filtering, and further moves them toward the reference ligand retrieved from structural databases via embedding-space mixup to create more challenging ranking tasks. A ranking objective then places greater emphasis on suppressing high-scoring invalid candidates that could contaminate the top-ranked screening results, providing structured supervision for lightweight adaptation. Experiments across diverse benchmarks demonstrate that this lightweight, pocket-specific adaptation outperforms both pretrained and fully retrained baselines while updating only the LayerNorm parameters, which account for approximately $0.03\%$ of the full model.
