المفاهيم الأساسية
본 논문에서는 레이블링된 데이터와 레이블링되지 않은 데이터를 모두 활용하는 준지도 학습 방식과 대조 학습 및 이산 웨이블릿 변환을 결합하여 실제 환경에서 촬영된 이미지의 헤이즈를 효과적으로 제거하는 새로운 네트워크인 WTCL-Dehaze를 제안합니다.
الملخص
WTCL-Dehaze: 웨이블릿 변환 및 대조 학습을 통한 실제 이미지 디헤이징 재고
본 연구에서는 단일 이미지 디헤이징을 위해 합성 데이터셋과 실제 데이터셋 간의 차이를 줄이고 다양한 환경에서 강력한 성능을 보장하는 향상된 준지도 학습 기반 디헤이징 네트워크를 개발하는 것을 목표로 합니다.
본 논문에서 제안된 WTCL-Dehaze는 대조 손실 및 이산 웨이블릿 변환(DWT)을 통합한 향상된 준지도 학습 기반 디헤이징 네트워크입니다.
주요 특징
대조 정규화: 헤이즈가 있는 이미지와 없는 이미지 쌍을 대조하여 특징 표현을 향상시켜 디헤이징 성능을 향상시킵니다.
웨이블릿 변환: 다중 스케일 특징 추출을 위해 DWT를 활용하여 고주파 디테일과 전역 구조를 효과적으로 캡처합니다.
준지도 학습: 레이블링된 데이터와 레이블링되지 않은 데이터를 모두 활용하여 도메인 차이를 완화하고 일반화 성능을 향상시킵니다.
네트워크 구조
인코더-디코더 구조, 스킵 연결, 정규화 레이어가 없는 잔차 블록으로 구성
인코더-디코더 구조의 가장 안쪽 레이어에 DWT를 통합하여 다중 스케일 특징 추출 수행
여러 잔차 블록을 통한 처리 후, 역 이산 웨이블릿 변환(IDWT)을 적용하여 웨이블릿 계수에서 특징 맵을 재구성
대립 학습을 위해 합성곱, 비선형 ReLU 및 인스턴스 정규화 레이어로 구성된 기존 분류기 아키텍처를 사용하여 판별기를 구성
학습 손실 함수
지도 손실: 평균 제곱 손실, 지각 손실, 적대적 손실
비지도 손실: 총 변동 손실, 어두운 채널 손실
대조 정규화(CR) 손실: 복원된 이미지를 깨끗한 이미지에 가깝게, 헤이즈가 있는 이미지에서 멀어지도록 유도