[논문 리뷰] Inverse design of dissipative quantum steady-states with implicit differentiation

Rodrigo A. Vargas–Hernández, Ricky T. Q. Chen외 2명|arXiv (Cornell University)|2020. 11. 25.

Advanced Thermodynamics and Statistical Mechanics참고 문헌 19인용 수 5

한 줄 요약

이 논문은 열역학적 손실이 있는 양자 안정 상태의 해밀토니안 매개변수에 대한 기울기를 계산하기 위한 새로운 암묵 미분 방법을 제안한다. 이는 효율적인 역설계와 민감도 분석을 가능하게 한다. 암묵함수정리와 자동미분를 활용하여, 일정한 메모리로 기울기를 계산하고 높은 정확도를 달성하였으며, Redfield 이론을 적용한 스핀-보존 모델에서 유한차분 결과와 일치함을 검증하였다.

ABSTRACT

Inverse design of a property that depends on the steady-state of an open quantum system is commonly done by grid-search type of methods. In this paper we present a new methodology that allows us to compute the gradient of the steady-state of an open quantum system with respect to any parameter of the Hamiltonian using the implicit differentiation theorem. As an example, we present a simulation of a spin-boson model where the steady-state solution is obtained using Redfield theory.

연구 동기 및 목표

• 열린 양자 시스템에서 해밀토니안 매개변수에 대한 양자 안정 상태 기울기를 계산하기 위한 확장 가능하고 메모리 효율적인 방법을 개발하는 것.
• 시스템 매개변수 ε와 Δ에 대해 ⟨σz⟩와 같은 관측량의 기울기를 계산하여 양자 시스템의 역설계를 가능하게 하는 것.
• 안정 상태를 계산하는 데 사용되는 수치 해법에 의존하지 않는 프레임워크를 제공하여 유연성과 광범위한 적용 가능성을 높이는 것.
• 안정 상태 성질이 시스템 매개변수에 의존하는 양자 시스템에서 민감도 분석과 최적화를 지원하는 것.
• Redfield 이론에 기반한 스핀-보존 모델을 통해 방법의 정확성과 효율성을 입증하는 것.

제안 방법

• 해밀토니안 매개변수에 대한 안정 상태 밀도 행렬의 기울기 ∂ρ^ss/∂θ_i를 암묵함수정리에 따라 계산한다. 이때 f(ρ^ss, θ) = 0 이 안정 상태 조건을 정의하며, ∂ρ^ss/∂θ_i = -[∂f/∂ρ]^(-1) [∂f/∂θ_i] 로 표현된다.
• 벡터-자기적 곱의 계산을 효율적으로 하기 위해 보조 미분방정식 dy/dt = yJ - v 를 풀어, 자명한 역행렬 계산을 피한다.
• 자동미분 라이브러리(예: JAX)를 사용하여 저비용으로 벡터-자기적 곱을 계산함으로써 기울기 계산의 확장성을 확보한다.
• 스핀-보존 모델에서 안정 상태는 Redfield 이론을 사용하여 계산되었으며, 기울기는 유한차분 결과와 비교하여 검증되었다.
• 제안된 암묵 미분 기반 기울기 계산 방식을 사용하여 Adam 최적화 알고리즘을 적용하였다.
• 이 방법은 중간 ODE 해법 상태나 시간 역행 궤적을 저장하지 않기 때문에 일정한 메모리 사용을 보이며, 대규모 양자 시스템에 적합하다.

실험 결과

연구 질문

• RQ1 ODE 해법 궤적을 저장하지 않고도 손실이 있는 양자 안정 상태의 기울기를 시스템 매개변수에 대해 효율적이고 정확하게 계산할 수 있는가?
• RQ2 기울기 계산에서 암묵 미분 방법이 안정 상태 관측량의 기울기에 대해 유한차분과 비교해 정확도가 어떻게 되는가?
• RQ3 이 방법을 통해 목표 안정 상태 관측량을 달성하기 위해 매개변수를 최적화함으로써 효과적인 양자 시스템의 역설계가 가능한가?
• RQ4 시스템 매개변수 ε와 Δ의 변화에 대해 안정 상태 기대값 ⟨σz⟩는 얼마나 민감한가?
• RQ5 안정 상태가 Redfield 이론과 같은 복잡한 many-body 이론을 통해 계산되는 시스템에도 이 방법이 확장 가능하고 적용 가능한가?

주요 결과

• 암묵함수정리를 활용하여 해밀토니안 매개변수에 대한 안정 상태 밀도 행렬의 기울기를 정확하게 계산하였으며, 결과가 유한차분 계산과 일치함을 확인하였다.
• 스핀-보존 모델에서 테스트한 매개변수 범위 내에서 기울기의 상대 오차가 10^-5 이하로 매우 높은 정확도를 달성하였다.
• 기대값 ⟨σz⟩는 Δ 증가에 따라 증가하는 경향을 보였으며, 시스템의 환경 매개변수 β, η, wc 변화에 대해 테스트 범위 내에서 민감도가 낮았다.
• Adam 최적화기법을 사용한 역설계는 목표 ⟨σz⟩ ≈ 0.04995847 을 재현하는 데 성공적으로 수렴하였으며, 100회 반복 후 손실 값은 2×10^-5 이하였다.
• 이 방법은 ODE 궤적 저장이나 시간 역행을 피함으로써 일정한 메모리 사용을 보였으며, 대규모 양자 시스템에 적합함을 입증하였다.
• 이 프레임워크는 초기 밀도 행렬에 대해서도 최적화를 가능하게 하여, 초기 조건 민감도 분석에의 적용 범위를 확장한다.