DGIST 로봇및기계전자공학과 박상현 교수팀이 의료영상에 특화된 도메인 적응 인공지능 모델 기술을 개발했다.

왼쪽부터 DGIST 박상현 교수, 강명균 박사과정생
왼쪽부터 DGIST 박상현 교수, 강명균 박사과정생

의료 인공지능의 경우 학습에 사용된 이미지 양식과 다른 이미지가 모델에 입력되면 성능이 크게 하락하는 일이 빈번하게 일어난다.

하지만 해당 기술은 서로 다른 스캐너나 다른 환경에서 취득한 의료영상이더라도 성능이 저하되지 않고 정확한 분석이 가능한 기술로, 영상의 특성이 바뀔 때마다 새롭게 학습데이터를 구축하고 레이블을 제작했던 기존의 방식보다 시간과 비용이 절감되어 의료 분야에 상당한 기여를 할 것으로 보인다.

또한, CT와 MRI 등 다중 모달리티 영상을 취득하기 어려운 경우에도 CT 영상으로부터 MRI로 변환한 데이터셋을 생성해 더 정확한 분석을 할 수 있게 됐다.

박상현 교수팀은 이미지 변환 작업을 진행할 때 ‘상호 정보 오류함수’를 활용하여 이미지 구조변형을 최소화할 수 있는 기술을 개발했다.

연구팀이 개발한 기술은 의료영상 이미지의 구조 정보와 새로운 도메인의 이미지에서 질감 정보를 추출하고, ‘판별자 오류함수’를 활용하여 사실적인 이미지를 생성한다.

그리고 새로운 도메인의 질감 정보를 유지하면서 구조변형이 적은 이미지를 생성하기 위해 각각 ‘질감 동시 발생 오류함수’ 와 ‘상호 정보 오류함수’를 활용해 새로운 도메인의 이미지를 생성했고 이렇게 생성된 이미지들을 함께 이용해 딥러닝 모델을 학습하면 도메인 적응이 가능해진다.

연구팀은 해당 기술을 활용하여 여러 기관에서 수집한 안저 영상 및 전립선 MRI, 심장 CT와 MRI 영역화를 위해 서로 다른 모달리티 이미지를 각각 반대로 생성하게 하여 도메인 적응을 수행했다.

그 결과 해당 기술을 통해 구조를 유지하면서 모달리티가 다른 이미지를 잘 생성할 수 있음을 확인했고, 기존의 도메인 적응 및 이미지 변환 기법과 비교했을 때 보다 우수한 성능을 나타낸다는 것을 확인했다.

박상현 교수는 “이번 연구를 통해 의료 분야에서 도메인이 바뀔 때마다 새롭게 인공지능 모델을 학습하는 데 드는 시간과 비용을 획기적으로 줄일 수 있는 기술을 개발했다”며 “해당 기술이 여러 의료현장에서 범용적으로 활용가능한 진단소프트웨어 개발에 크게 기여할 것으로 기대한다”고 말했다.

산업통상자원부의 현장수요의료기기고도화기술개발 사업과 한국연구재단의 신진연구자지원 사업의 지원을 받아 수행한 이번 연구결과는 영상분석 관련 분야 국제학술지 ‘Pattern Recognition’에 게재됐다.

저작권자 © 코리아헬스로그 무단전재 및 재배포 금지