이 연구에서는 Mamba 아키텍처를 활용한 새로운 확산 모델 MD-Dose를 제안한다. MD-Dose는 전방 확산 과정과 역방향 복원 과정으로 구성된다. 전방 과정에서는 선량 분포도에 점진적으로 가우시안 노이즈를 추가하여 순수 노이즈 이미지를 생성한다. 역방향 과정에서는 Mamba 기반 노이즈 예측기를 활용하여 노이즈를 예측하고, 최종적으로 선량 분포도를 출력한다. 또한 Mamba 인코더를 개발하여 해부학적 정보를 추출하고, 이를 노이즈 예측기에 통합함으로써 표적 체적(PTV) 및 위험 장기(OAR) 영역의 선량 분포를 더 정확하게 예측할 수 있다. 300명의 흉부 종양 환자 데이터셋에 대한 실험 결과, MD-Dose가 다양한 지표와 시간 소비 측면에서 우수한 성능을 보였다.
A otro idioma
del contenido fuente
arxiv.org
Ideas clave extraídas de
by Linjie Fu,Xi... a las arxiv.org 03-14-2024
https://arxiv.org/pdf/2403.08479.pdfConsultas más profundas