챕터별 상세
솔루션 개요 및 전체 파이프라인
전체 프로세스는 딥러닝 모델을 통한 1차 속도 추정, 데이터 스타일에 따른 사전 확률(Prior) 선택, 그리고 최종적인 베이지안 FWI 최적화 단계로 구성된다. 딥러닝 모델은 Brendan Artley가 공개한 CAFormer를 그대로 활용하여 초기 속도 프로파일을 생성했다. 이후 데이터의 지질학적 특성에 따라 Gaussian Process나 Total Variation 중 적합한 Prior를 선택하여 물리적 정합성을 높였다.
초기값 생성을 위한 딥러닝 모델: CAFormer
초기 속도 프로파일을 얻기 위해 Vision Transformer 기반의 CAFormer 모델을 사용했다. 이 모델은 Cross-attention, Pixel shuffle, Squeeze-and-excitation 기법이 적용되어 이미지 형태의 속도 지도를 효과적으로 추정한다. 발표자는 딥러닝 모델 자체의 개선보다는 이후 단계인 베이지안 정제 과정이 성능 향상에 더 결정적이라고 판단하여 공개된 모델을 튜닝 없이 사용했다.
python
c(x) = p(x) + ||s(x) - m||^2베이지안 역산에서 최소화하고자 하는 비용 함수로, 사전 확률(Prior)과 관측 데이터 오차(Residual)의 합으로 구성된다.
데이터 스타일별 사전 확률(Prior) 설계
지하 구조의 형태에 따라 서로 다른 수학적 Prior를 적용했다. 매끄러운 속도 변화를 보이는 Style A 데이터에는 Gaussian Process Prior를 적용하여 부드러운 전이를 유도했다. 반면 지층 경계가 뚜렷한 FlatVel이나 Style B 데이터에는 Total Variation(TV) Prior를 사용하여 급격한 속도 변화를 보존하면서 노이즈를 억제했다. 이러한 맞춤형 Prior 설계가 복원 정확도를 크게 향상시켰다.
베이지안 FWI 최적화 및 솔버 구현
비용 함수 최소화를 위해 BFGS와 Gauss-Newton 알고리즘을 혼합하여 사용했다. BFGS 구현 시 히스토리 크기를 제한하지 않고 전체 파라미터 범위를 활용하여 수렴 성능을 극대화했다. 또한 깊이에 따른 사전 조건화(Preconditioning)를 시도했으나 실질적인 이득이 없어 최종 솔루션에서는 제외했다. 모든 최적화 과정은 물리적 파동 방정식을 만족하도록 설계되었다.
커스텀 CUDA 커널을 통한 연산 가속
대규모 역산 계산의 효율성을 위해 직접 작성한 CUDA 커널을 도입했다. PyTorch의 기본 연산 대신 수동으로 구현한 커널을 통해 순전파(Forward modeling)는 30ms, 역전파(Gradients)는 70ms 수준으로 단축했다. 이를 통해 V100 GPU 한 장으로도 65,000개의 데이터셋을 약 700달러의 비용으로 처리할 수 있는 경제성을 확보했다.
용어 해설
- 완전 파형 역산(Full Waveform Inversion)
- — 지진파나 음파의 전체 파형 데이터를 물리적 파동 방정식과 비교하여 지하 매질의 속도나 밀도 같은 물성을 고해상도로 추정하는 기술이다. 물리 기반 시뮬레이션과 관측 데이터 사이의 오차를 최소화하는 비선형 최적화 과정을 거치며 지질 탐사에서 핵심적인 역할을 한다.
- 가우시안 프로세스(Gaussian Process)
- — 데이터의 분포를 무한 차원의 가우시안 분포로 가정하여 함수 자체를 확률적으로 모델링하는 비모수적 베이지안 기법이다. 평균과 커널 함수를 통해 데이터의 매끄러움이나 주기성을 정의하며, 예측값과 함께 불확실성을 동시에 제공하는 장점이 있다.
- 전체 변동성(Total Variation)
- — 신호나 이미지에서 인접한 지점 간의 값 차이의 절대값 합을 의미하며, 이를 최소화하는 정규화 기법으로 주로 사용된다. 노이즈를 억제하면서도 지층 경계와 같은 급격한 불연속면(Sharp edges)을 보존하는 특성이 있어 역산 문제에서 유용하다.
- BFGS 알고리즘(BFGS)
- — 함수의 최솟값을 찾기 위해 2차 미분값인 헤시안 행렬을 직접 계산하는 대신 이전 단계의 기울기 변화를 통해 이를 근사적으로 업데이트하는 Quasi-Newton 최적화 기법이다. 메모리 효율성과 수렴 속도가 뛰어나 대규모 파라미터 최적화 문제에 널리 쓰인다.
AI 분석 전체 내용 보기
AI 요약 · 북마크 · 개인 피드 설정 — 무료
출처 · 인용 안내
원문 발행 2026. 02. 21.수집 2026. 02. 21.출처 타입 YOUTUBE
인용 시 "요약 출처: AI Trends (aitrends.kr)"를 표기하고, 사실 확인은 원문 보기 기준으로 진행해 주세요. 자세한 기준은 운영 정책을 참고해 주세요.
