แนวคิดหลัก
본 논문에서는 GAN 기반 모델과 폐쇄형 인수분해를 활용하여 사실적인 피부 병변 이미지를 합성 생성하고, 이를 통해 피부암 진단 모델의 성능을 향상시키는 새로운 방법론을 제시합니다.
บทคัดย่อ
피부 병변 이미지 합성 생성 및 분류 모델 성능 향상 연구 논문 요약
인공지능 기반 피부암 진단 모델의 정확도와 일반화 가능성을 높이기 위해서는 다양하고 고품질의 주석 이미지 데이터 세트가 필수적입니다.
그러나 실제 의료 데이터 부족과 개인정보 보호 문제로 인해 다양한 피부 병변 이미지 데이터를 수집하는 데 어려움을 겪고 있습니다.
본 논문에서는 GAN 기반 모델과 폐쇄형 인수분해를 활용하여 사실적인 피부 병변 이미지를 합성 생성하고, 이를 통해 피부암 진단 모델의 성능을 향상시키는 새로운 방법론을 제시합니다.
StyleGAN2 모델 학습:
다양한 피부 병변 이미지 데이터 세트(HAM10000, Fitzpatrick, Seven-Point Checklist Dermatology)를 사용하여 고품질 합성 이미지를 생성하도록 StyleGAN2 모델을 학습합니다.
잠재 공간 인수분해를 통한 의미론적 방향 추출:
StyleGAN2 모델의 잠재 공간에서 의미 있는 잠재적 의미론적 방향(예: 크기, 질감, 기하학적 특성, 배경 특성)을 식별하기 위해 폐쇄형 인수분해를 사용합니다.
HyperStyle을 활용한 GAN 반전:
실제 이미지를 GAN의 잠재 공간에 매핑하기 위해 인코더 및 옵티마이저 유닛으로 구성된 HyperStyle 기반 GAN 반전 모델을 학습합니다.
관련 변형 식별:
폐쇄형 인수분해를 통해 추출된 의미론적 방향을 검증하고 개선하여 사람이 이해할 수 있고 도메인과 관련된 변형만 고려합니다.
합성 데이터를 활용한 분류기 성능 향상:
합성 이미지를 사용하여 원본 데이터 세트를 증강하고, DenseNet121 및 DenseNet169 아키텍처를 기반으로 하는 피부 병변 분류 모델을 학습합니다.
증강된 데이터 세트로 학습된 모델의 성능을 평가하고 기준 모델과 비교합니다.