KPPax: 대기 화학 분야 머신러닝을 위한 미분 가능 운동학적 ODE 전처리

요약

KPPax는 대기 화학 메커니즘을 JAX 기반의 미분 가능 코드로 변환하는 전처리 프로그램입니다. 이 도구는 적절 변환(adjoints)과 로그 공간 희소 연산을 활용하여 수치적으로 정확한 민감도를 계산합니다. 이를 통해 복잡한 물리-화학 모델에 매개변수 추정 및 신경망 보정과 같은 머신러닝 기법을 직접 적용할 수 있게 합니다.

핵심 포인트

• KPPax는 KPP 메커니즘을 JAX에서 미분 가능하고 벡터화 가능한 코드로 변환합니다.
• 적절 변환(adjoints)을 이용해 복잡한 시스템의 정확한 민감도 계산이 가능합니다.
• Fisher 정보 행렬 등을 활용하여 반응 계수의 불확실성을 정량적으로 분석할 수 있습니다.
• 신경망 보정치를 통해 화학 모델의 생물 발생성 오류를 크게 감소시킬 수 있습니다.