2025-11-16T00:34:12.699199

Latent Retrieval Augmented Generation of Cross-Domain Protein Binders

Zhang, Kong, Huang et al.

Designing protein binders targeting specific sites, which requires to generate realistic and functional interaction patterns, is a fundamental challenge in drug discovery. Current structure-based generative models are limited in generating nterfaces with sufficient rationality and interpretability. In this paper, we propose Retrieval-Augmented Diffusion for Aligned interface (RADiAnce), a new framework that leverages known interfaces to guide the design of novel binders. By unifying retrieval and generation in a shared contrastive latent space, our model efficiently identifies relevant interfaces for a given binding site and seamlessly integrates them through a conditional latent diffusion generator, enabling cross-domain interface transfer. Extensive exeriments show that RADiAnce significantly outperforms baseline models across multiple metrics, including binding affinity and recovery of geometries and interactions. Additional experimental results validate cross-domain generalization, demonstrating that retrieving interfaces from diverse domains, such as peptides, antibodies, and protein fragments, enhances the generation performance of binders for other domains. Our work establishes a new paradigm for protein binder design that successfully bridges retrieval-based knowledge and generative AI, opening new possibilities for drug discovery.

academic

잠재 검색 증강 생성을 통한 교차 도메인 단백질 결합자 설계

기본 정보

논문 ID: 2510.10480
제목: Latent Retrieval Augmented Generation of Cross-Domain Protein Binders
저자: Zishen Zhang, Xiangzhe Kong, Wenbing Huang, Yang Liu
분류: cs.LG cs.AI
발표 시간/학회: 사전인쇄본. 검토 중 (2024년 10월)
논문 링크: https://arxiv.org/abs/2510.10480

초록

특정 위치의 단백질 결합자 설계는 약물 발견의 기본적인 과제로, 현실적이고 기능적인 상호작용 패턴의 생성이 필요합니다. 현재의 구조 기반 생성 모델은 충분한 합리성과 해석 가능성을 갖춘 인터페이스 생성에 제한이 있습니다. 본 논문은 알려진 인터페이스를 활용하여 새로운 결합자 설계를 지도하는 검색 증강 확산 정렬 인터페이스 프레임워크(RADiAnce)를 제안합니다. 공유 대조 잠재 공간에서 검색과 생성을 통합함으로써, 이 모델은 주어진 결합 위치에 대한 관련 인터페이스를 효율적으로 식별하고, 조건부 잠재 확산 생성기를 통해 원활하게 통합하여 교차 도메인 인터페이스 전이를 실현합니다.

연구 배경 및 동기

핵심 문제

단백질 결합자 설계 과제: 특정 단백질 위치를 표적으로 하는 결합자 설계는 현실적이고 기능적인 분자 인터페이스 상호작용 패턴의 생성을 요구합니다
기존 방법의 한계: 현재 구조 생성 모델은 합리성과 해석 가능성이 부족하며, 알려진 구조 정보를 효과적으로 활용하지 못합니다

중요성

약물 발견, 구조 생물학 등 다양한 분야에서 광범위한 응용 가치 보유
전통적 방법은 물리적 또는 통계적 에너지 경관 샘플링 최적화에 의존하여 효율성이 낮음
심층 생성 모델은 진전이 있지만 여전히 합리적인 분자 인터페이스 생성이 어려움

기존 방법의 한계

사전 지식 무시: 대부분의 방법은 목표 결합 위치만을 기반으로 생성하며, 기존 단백질 복합체의 풍부한 재사용 가능한 상호작용 패턴을 무시합니다
교차 도메인 일반화 부족: 펩타이드, 항체, 단백질 단편 등 다양한 유형의 결합자 간 공통 상호작용 모티프를 효과적으로 활용하지 못합니다
해석 가능성 부족: 생성 과정에 명확한 생물학적 지도 원칙이 부재합니다

핵심 기여

RADiAnce 프레임워크 제안: 단백질 결합자 서열-구조 협력 설계에 검색 증강 생성을 적용한 첫 번째 방법
대조 잠재 공간 구축: 검색과 생성을 통합하는 공유 잠재 표현을 설계하여 교차 도메인 인터페이스 유사성 측정 지원
교차 도메인 인터페이스 전이 실현: 다양한 결합자 유형의 인터페이스 검색이 다른 도메인 결합자의 생성 성능을 향상시킬 수 있음을 검증
현저한 성능 향상: 결합 친화력, 기하학적 형태 및 상호작용 복구를 포함한 여러 평가 지표에서 기준 방법을 크게 능가

방법론 상세 설명

작업 정의

입력: 목표 단백질의 결합 위치 Y (10Å 거리 내의 잔기)
출력: 해당 위치와 특이적으로 결합할 수 있는 분자 결합자 X
목표: 조건부 분포 p_θ(X | Y, T(Y|D)) 모델링, 여기서 T(Y|D)는 데이터베이스 D에서 검색된 관련 인터페이스

모델 아키텍처

1. 대조 변분 자동인코더(Contrastive VAE)

인코더: Zx = Eφ(X), Zy = Eφ(Y)
디코더: X̂ = Dξ(Zx, Zy, Y)

주요 설계:

결합 위치 Y와 결합자 X를 독립적으로 잠재 포인트 클라우드로 인코딩
잠재 변수는 스칼라 임베딩 zi와 3D 좌표 z⃗i 포함
대조 학습을 통해 양성 샘플 쌍을 정렬하고 음성 샘플 쌍을 배제

손실 함수:

L(D) = Σ(Lrec + LKL + Lretrieval)

여기서:

Lrec: 재구성 손실 (교차 엔트로피 + MSE)
LKL: KL 발산 정규화
Lretrieval: 양방향 대조 손실

2. 검색 증강 잠재 확산

정방향 과정:

q(u⃗ti | u⃗t-1i) = N(u⃗ti; √(1-βt)·u⃗t-1i, βtI)

역방향 과정:

pθ(u⃗t-1i | Ztx, Zy, Tv) = N(u⃗t-1i; μ⃗θ(Ztx, Zy, Tv), βtI)

템플릿 통합 메커니즘:

E(3) 등변 Transformer를 노이즈 제거 핵심으로 사용
교차 주의 메커니즘을 통해 검색된 템플릿 정보 통합
쿼리-키-값 계산: Q = HWQ, K = TWK, V = TWV

기술 혁신점

통합 잠재 공간: 동일한 잠재 공간에서 검색과 생성을 통합하여 검색 결과가 생성 과정을 직접 지도할 수 있도록 보장
교차 도메인 유사성 측정: 대조 학습을 통해 학습된 잠재 표현이 다양한 유형의 결합자 간 공통 상호작용 모티프를 포착할 수 있음
조건부 확산 통합: 검색된 인터페이스 임베딩을 교차 주의 및 잔여 MLP를 통해 확산 과정에 혁신적으로 통합

실험 설정

데이터셋

펩타이드 설계: PepBench 데이터셋
- 훈련: 4,157개 복합체
- 검증: 114개 복합체
- 테스트: 93개 LNR 벤치마크 사례
항체 설계: SAbDab 데이터셋
- 훈련: 9,473개 항목
- 검증: 400개 항목
- 테스트: 60개 RAbD 벤치마크 사례
단백질 단편: ProtFrag 데이터셋
- 70,498개 단량체 유래 단백질 단편

평가 지표

AAR (아미노산 복구율): 생성된 서열과 참조 서열의 일치 비율
RMSD: Cα 좌표의 제곱 평균 제곱근 편차
ISM (상호작용 위치 일치): 주요 물리화학 상호작용의 복구 정도
∆∆G: 결합 자유 에너지 변화
IMP: 생성된 결합자가 천연 리간드를 능가하는 목표 비율

비교 방법

펩타이드 설계: RFDiffusion, PepFlow, PepGLAD, UniMoMo
항체 설계: MEAN, DyMEAN, DiffAb, GeoAB, UniMoMo

모델	AAR (%)	RMSD (Å)	∆∆G (kJ/mol)	IMP (%)	ISM (%)
RFDiffusion	34.68	4.69	24.78	5.38	28.38
PepFlow	35.47	2.87	15.71	14.13	27.83
PepGLAD	38.62	2.74	15.26	16.13	32.63
UniMoMo	38.69	2.31	2.409	40.86	49.13
RADiAnce	39.42	2.29	1.963	41.94	52.15