核心概念
본 논문에서는 콘빔 CT(CBCT)에서 발생하는 움직임으로 인한 이미지 품질 저하 문제를 해결하기 위해 그래디언트 기반 최적화 알고리즘을 사용한 새로운 모션 추정 접근 방식을 제안합니다. 이 접근 방식은 기존 방법보다 19배 빠른 속도를 제공하며, 보다 정확한 모션 추정을 위해 픽셀 단위 품질 맵을 예측하는 자동 인코더와 유사한 아키텍처를 사용합니다.
要約
콘빔 CT에서 빠르고 정확한 헤드 모션 보정을 위한 그래디언트 기반 접근 방식
콘빔 컴퓨터 단층 촬영(CBCT) 시스템은 유연성이 뛰어나 직접적인 현장 진료 영상, 특히 급성 뇌졸중 평가와 같은 중요한 상황에서 유망한 방법으로 주목받고 있습니다. 그러나 CBCT를 임상 workflow에 통합하는 데에는 어려움이 따르는데, 주로 스캔 시간이 길어 스캔 중 환자의 움직임이 발생하고 재구성된 볼륨의 이미지 품질이 저하되는 것과 관련이 있습니다.
본 논문에서는 콘빔 CT 기하학적 구조에 대한 역투영 연산자의 일반화된 미분을 활용하는 그래디언트 기반 최적화 알고리즘을 사용하여 CBCT 모션 추정에 대한 새로운 접근 방식을 제시합니다. 이를 기반으로 재구성 공간에서 현재 모션 추정의 품질을 평가하는 완전히 미분 가능한 목표 함수를 공식화합니다.
그래디언트 기반 최적화 파이프라인
본 논문에서 제안하는 모션 추정을 위한 자동 초점 기반 목표 함수는 모션 매개변수에 대한 유익한 그래디언트 업데이트를 생성합니다. 목표 함수는 투영 데이터와 스캔 기하학적 구조의 초기화로서 보정된 투영 행렬 집합에만 의존합니다. 전체 목표 함수 f(x) : RN →R은 그림 2에 강조 표시된 세 가지 별개의 처리 단계로 구성됩니다.
모션 모델: 모션 궤적을 설명하는 자유 매개변수 x ∈RN은 모션 모델 p : RN →RM에 의해 처리되어 초기 보정 행렬과 x로 설명되는 모션 패턴에서 투영 행렬을 계산합니다. 여기서 M은 스캔의 모든 투영 행렬의 항목 수입니다.
역투영: 역투영 r : RM →RK는 업데이트된 투영 행렬을 사용하여 필터링된 투영 데이터를 K개 요소가 있는 볼륨으로 분석적으로 역투영합니다. 단순화를 위해 필터링된 투영 데이터는 역투영 함수 r에 대한 입력으로 생략합니다. 역투영은 현재 자유 매개변수 추정값이 주어진 데이터의 중간 재구성을 생성합니다.
품질 메트릭: 품질 메트릭 q : RK →R은 중간 재구성이 주어진 스칼라 품질 값을 반환합니다. 본 논문에서 제안하는 방법에서는 먼저 입력 볼륨과 동일한 차원의 전체 품질 맵을 회귀 분석한 다음 평균을 내어 단일 값으로 만듭니다.
픽셀 단위 품질 메트릭 회귀
역투영에서 중간 재구성이 주어지면 모션 아티팩트 측면에서 품질을 정량화하고자 합니다. 추론 시 이 매핑은 참조가 없어야 합니다. 즉, 모션에 영향을 받는 볼륨만 주어지고 모션이 없는 참조는 주어지지 않은 상태에서 품질을 추정해야 합니다. 이 작업을 해결하기 위해 모션에 영향을 받는 스캔과 해당 모션이 없는 스캔의 시뮬레이션된 쌍 데이터 세트에서 신경망을 훈련하여 모션에 영향을 받는 볼륨만 주어진 참조 기반 대상 메트릭에 근접합니다. 이는 이전 연구에서 유사하게 제안되었습니다[6], [20], [23].
기존 접근 방식과 달리 훈련된 네트워크를 그래디언트 기반 설정에서 사용합니다. 여기서 네트워크 출력의 입력 볼륨에 대한 미분이 모션 추정을 위해 계산됩니다. 그래디언트가 없는 방법의 경우 입력 볼륨에서 품질 메트릭을 정확하게 회귀 분석하는 것이 중요합니다. 그래디언트 기반의 경우 훈련된 네트워크를 통해 역전파를 통해 얻은 유익한 볼륨 그래디언트는 순방향 매핑만큼 중요합니다. 유익한 볼륨 그래디언트를 얻을 수 있도록 공간적으로 해결된 전체 품질 맵을 회귀 분석하는 볼륨-대-볼륨 네트워크를 훈련할 것을 제안합니다. 이는 계약 아키텍처를 사용하여 스칼라 품질 메트릭을 직접 회귀 분석하는 기존 접근 방식과 대조적입니다. 최적화를 위해 스칼라를 얻기 위해 체적 품질 맵의 평균을 구하여 각 공간 위치가 최종 값에 동일하게 기여하도록 합니다.
그래디언트 기반 최적화 알고리즘
예측된 품질 맵의 평균을 최소화하여 스캔별 모션 추정값 x∗를 얻습니다. 중요한 것은 최적화 시 품질 메트릭 네트워크의 매개변수가 완전히 훈련되고 고정된다는 것입니다. 기존 모션 보정 알고리즘의 대부분은 그래디언트가 없는 최적화 프로그램을 사용하여 모션 매개변수와 관련하여 목표를 최소화합니다. 그래디언트가 없는 최적화 알고리즘의 경우 목표 함수만 지정하면 되는 반면, 그래디언트 기반 알고리즘의 경우 목표 값의 모션 매개변수에 대한 의존성이 명확하지 않기 때문에 그래디언트 표현식을 공식화하기 어려울 수 있습니다[37]. 본 논문에서는 그래디언트 기반 최적화를 적용할 수 있는 완전히 미분 가능한 자동 초점 대상 함수를 처음으로 사용합니다. 구체적으로, 다음과 같은 형태의 기본적인 그래디언트 하강 업데이트 체계를 사용합니다.
x(n+1) = x(n) −s(n) · df/dx.
업데이트된 추정값 x(n)은 스텝 크기 s(n) ∈R로 조정된 음의 그래디언트 방향으로 스텝을 수행하여 초기 추정값 x(0) = 0에서 반복적으로 계산됩니다. 스텝 크기는 s(n) = s0 · tn이 되도록 지수적으로 감소하며, 여기서 t ∈R은 감쇠 계수이고 s0 ∈R은 초기 스텝 크기를 나타냅니다. 달리 명시되지 않는 한, 초기 스텝 크기 s0 = 100, 감쇠 계수 t = 0.97로 100회 반복하여 알고리즘을 실행합니다.