[논문 리뷰] Computational Anatomy in Theano

Line Kühnel, Stefan Sommer|arXiv (Cornell University)|2017. 06. 15.

Morphological variations and asymmetry인용 수 1

한 줄 요약

이 논문은 계산 해부학에서 미분 기하학과 비선형 통계를 위한 기호적 고성능 구현을 Theano를 사용해 제시한다. 지오데식선, 평행 이동, 프레셰 평균, 브라운 운동을 간결하게 GPU 가속으로 계산할 수 있으며, 20,000개의 랜드마크를 사용한 Corpus Callosum 형태에 대해 LDDMM 프레임워크를 적용하여 밀리초 이내의 시간으로 성능을 입증하였다.

ABSTRACT

To model deformation of anatomical shapes, non-linear statistics are required to take into account the non-linear structure of the data space. Computer implementations of non-linear statistics and differential geometry algorithms often lead to long and complex code sequences. The aim of the paper is to show how the Theano framework can be used for simple and concise implementation of complex differential geometry algorithms while being able to handle complex and high-dimensional data structures. We show how the Theano framework meets both of these requirements. The framework provides a symbolic language that allows mathematical equations to be directly translated into Theano code, and it is able to perform both fast CPU and GPU computations on high-dimensional data. We show how different concepts from non-linear statistics and differential geometry can be implemented in Theano, and give examples of the implemented theory visualized on landmark representations of Corpus Callosum shapes.

연구 동기 및 목표

• 계산 해부학에서 복잡한 미분 기하학 및 비선형 통계 알고리즘의 구현을 단순화하기 위해.
• Theano의 기호적 미분 및 GPU/CPU 계산을 활용하여 고차원 해부학 데이터를 효율적으로 처리하기 위해.
• 기반 랜드마크의 다양체 분석에서 리만 및 부분 리만 기하학적 구조에 대해 프레임워크를 시연하기 위해.
• 프레임 번들의 활용을 통해 비선형 다양체에서의 프레셰 평균 및 정규 분포를 정확하게 추정하기 위해.
• 비선형 데이터 공간에서의 비선형 통계 분석을 위한 일반 목적의 재사용 가능한 코드베이스 제공하기 위해.

제안 방법

• Theano의 자동 미분를 활용한 기호적 지오데식선 계산을 위한 하미льт로니안 방정식의 기호적 구현.
• 메트릭 텐서 위에서 기호적 행렬 연산을 통한 크리스토펠 기호 및 평행 이동 계산.
• 다양체 및 프레임 번들의 프레셰 평균 알고리즘을 사용하여 지오데식 거리와 최대 가능 경로(MPP) 에너지를 최소화.
• 방정식 (6)의 SDE 해를 사용한 다양체 상의 브라운 운동의 확률적 개발을 통해 전이 분포를 정의하고 정규 분포를 생성.
• Theano를 통한 GPU 가속 통합을 통해 명시적인 GPU 코드 없이도 자동으로 작동.
• 운동량 벡터와 날카운 맵을 사용하여 프레임 번들의 프레셰 평균을 기반으로 한 MPP 정의를 위한 구현.

실험 결과

연구 질문

• RQ1 Theano의 기호적 프레임워크는 계산 해부학에서 복잡한 미분 기하학 알고리즘의 구현을 단순화할 수 있는가?
• RQ2 Theano는 고차원 랜드마크 다양체에서 지오데식선과 평행 이동을 얼마나 효율적으로 계산할 수 있는가?
• RQ3 프레임 번들의 프레셰 평균은 비선형 형태 다양체에서 이방형 정규 분포를 효과적으로 추정하는 데 사용될 수 있는가?
• RQ4 Theano는 명시적인 GPU 프로그래밍 없이 얼마나 높은 수준의 GPU 가속을 가능하게 하는가?
• RQ5 구현된 확률적 과정, 예를 들어 브라운 운동은 비선형 다양체에서 유클리드 통계의 유사물과 비교해 어떻게 성능을 발휘하는가?

주요 결과

• Theano를 통해 고차원 도함수의 수작업 유도 없이도 복잡한 미분 기하학 방정식을 직접 기호적으로 번역하여 간결하고 이해하기 쉬운 코드로 변환할 수 있었다.
• 프레임워크는 20,000개의 점을 가진 40,000차원 랜드마크 다양체에서 지오데식 매칭을 위해 CPU 또는 GPU를 투명하게 사용하여 밀리초 이내의 계산 시간을 달성하였다.
• 메트릭 텐서 위에서 기호적 행렬 연산을 통해 평행 이동과 크리스토펠 기호 계산이 성공적으로 구현되었다.
• 프레임 번들의 프레셰 평균이 성공적으로 계산되어, 수직이 아닌 기저를 사용한 이방형 정규 분포 추정이 가능해졌다.
• 브라운 운동과 정규 분포 전이가 Corpus Callosum 다양체에서 시각화되었으며, 이는 곡면에서의 확률 모델의 타당성을 확인하였다.
• 구현은 일반 목적의 것으로서 재사용 가능했으며, Theano Geometry 레포지터리에서 코드를 공개하여 랜드마크 다양체를 초월한 응용에 활용할 수 있도록 하였다.