TxAgent: 치료 추론 및 맞춤형 약물 치료를 위한 AI 에이전트 192

TxAgent란 무엇인가요?

TxAgent는 하버드 의과대학에서 개발한 인공지능 치료 에이전트로, 다단계 추론과 실시간 생물의학 지식 검색을 결합하여 211개의 도구로 구성된 툴킷을 활용해 환자에게 개인화된 치료 제안을 제공합니다. 약물 상호작용, 금기 사항을 분석하고, 환자의 개별 특성(연령, 유전적 요인, 질병 진행 등)에 따라 치료 전략을 수립할 수 있습니다.

TxAgent 기능

• 다단계 추론: TxAgent는 분자, 약동학 및 임상 수준에서 약물 상호작용을 평가하고, 환자의 동반 질환 및 병용 약물에 따라 금기 사항을 식별할 수 있습니다.

• 실시간 지식 검색: 여러 생물의학 소스에서 증거를 검색하고 종합하며, 반복적 추론을 통해 치료 제안을 최적화합니다.

• 도구 호출 및 의사 결정 지원: TxAgent는 작업 목표에 따라 적절한 도구를 선택하고, 임상 추론과 다중 소스 검증이 필요한 치료 작업을 해결하기 위해 구조화된 함수 호출을 실행합니다.

• 개인 맞춤형 치료 계획: 환자의 연령, 유전적 요인, 질병 진행 등 개별 특성에 따라 치료 전략을 맞춤화할 수 있습니다.

• 투명한 추론 과정: TxAgent는 자연어 응답을 생성하고, 의사 결정 과정의 각 단계를 설명하는 상세한 추론 경로를 제공합니다.

TxAgent의 능력

• 지식 기반 앵커링: 도구 호출을 통해 검증된 지식을 확보하고 이를 바탕으로 출력을 생성합니다.

• 목표 지향적 도구 선택: ToolRAG 모델을 사용하여 ToolUniverse에서 가장 적합한 도구를 선택하고 적용합니다.

• 다단계 추론 문제 해결: 여러 번의 반복적 사고와 함수 호출을 통해 복잡한 작업을 해결하거나 도구의 예기치 않은 응답을 처리합니다.

• 지속적으로 갱신되는 지식 베이스 활용: 도구를 통해 지속적으로 업데이트되는 데이터베이스에 접근하여 TxAgent의 내장 지식 범위를 초과하는 문제를 해결합니다.

성능

• TxAgent는 3,168개의 약물 추론 작업과 456개의 개인 맞춤 치료 시나리오를 다루는 여러 벤치마크에서 뛰어난 성능을 보였습니다:

• 개방형 약물 추론 작업에서 정확도 92.1%를 달성하여 GPT-4o보다 25.8% 높은 성능을 보였습니다.

• 구조화된 다단계 추론에서 DeepSeek-R1(671B)을 능가했습니다.

• 약물명 변형 및 설명에 대한 일반화 능력에서 <0.01의 분산을 유지하여 기존 도구 사용 LLM을 55% 이상 능가했습니다.

TxAgent 활용 분야

• 약물 추론 작업: TxAgent는 3,168개의 약물 추론 작업과 456개의 개인 맞춤 치료 시나리오를 다루는 여러 벤치마크에서 뛰어난 성능을 보입니다.

• 임상 의사 결정 지원: 의사에게 정밀한 치료 제안을 제공하여 유해 사건의 위험을 줄이고 치료 의사 결정을 향상시킬 수 있습니다.

• 약물 상호작용 분석: 외부 데이터베이스와 머신러닝 모델을 호출하여 약물 간 상호작용을 평가합니다.

• 개인 맞춤 치료 조정: 예를 들어, 환자의 간 기능 상태에 따라 약물 용량을 조정합니다.

사용 방법

1. 의존성 설치:

H100 GPU 사용을 권장하며, 메모리는 80GB 이상이어야 합니다.

• ToolUniverse 설치:

pip install tooluniverse

• TxAgent 설치:

pip install txagent

2. 예제 실행:

• 예제 코드 실행:

python run_example.py

• Gradio 데모 실행:

python run_txagent_app.py

3. 사전 학습 모델:사전 학습된 모델 가중치는 HuggingFace에서 확인할 수 있습니다.

4. 툴박스(ToolUniverse):TxAgent의 핵심 구성 요소로, 1939년 이후 미국 FDA가 승인한 모든 약물을 포함하여 211개의 도구를 포함합니다.