Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

MedBayes-Lite: Bayesian Uncertainty Quantification for Safe Clinical Decision Support

Created by
  • Haebom
Category
Empty

저자

Elias Hossain, Md Mehedi Hasan Nipu, Maleeha Sheikh, Rajib Rana, Subash Neupane, Niloofar Yousefi

개요

MedBayes-Lite는 변환기 기반의 임상 언어 모델을 위한 경량 베이지안 향상 기술로, 신뢰할 수 있고 불확실성을 인지하는 예측을 생성하도록 설계되었습니다. 과도한 확신 문제를 해결하기 위해, 재훈련이나 아키텍처 변경 없이 기존 변환기 파이프라인에 불확실성 정량화를 직접 통합합니다. MedBayes-Lite는 (i) 에피스테믹 불확실성을 위한 몬테 카를로 드롭아웃을 사용하는 베이지안 임베딩 보정, (ii) 토큰 신뢰도를 통합하는 불확실성 가중치 어텐션, (iii) 임상 위험 최소화에서 영감을 얻은 신뢰도 기반 의사 결정 형성을 포함합니다. 생의학 QA 및 임상 예측 벤치마크(MedQA, PubMedQA, MIMIC-III)에서, MedBayes-Lite는 보정과 신뢰도를 지속적으로 개선하여 과도한 확신을 32~48% 감소시킵니다. 시뮬레이션된 임상 환경에서, 불확실한 예측을 검토하도록 표시하여 최대 41%의 진단 오류를 예방할 수 있습니다.

시사점, 한계점

신뢰할 수 있는 불확실성 전파를 가능하게 하고 의료 AI 시스템의 해석 가능성을 향상시킵니다.
과도한 확신을 감소시키고 예측의 신뢰도를 높입니다.
진단 오류 예방에 기여합니다.
추가적인 학습이나 아키텍처 변경이 필요하지 않습니다.
매개변수 오버헤드가 3% 미만으로 경량입니다.
👍