CosmosNews

#PyTorch

2건의 기사

천문·천체물리

GPU 가속 은하-은하 강한 중력렌즈 시뮬레이션 프레임워크 공개… CPU 대비 약 1,000배 속도

연구진이 PyTorch 텐서 기반의 GPU 가속 시뮬레이션 프레임워크를 제시해 고정밀 은하-은하 강한 중력렌즈 이미지를 생성했다. 합성 분광에너지분포(SED)를 결합해 렌즈·광원 은하의 적색편이 의존적 측광 특성을 모델링하고, 점확산함수(PSF)·한계등급·영점 등 관측 변수를 반영해 특정 서베이 조건을 재현함으로써 서베이 간 결합 분석을 가능하게 했다. 응용 사례로 동일한 렌즈 모델 변수를 써서 KiDS, LSST, 유클리드(Euclid)의 다중 대역 이미지를 시뮬레이션하고 딥러닝 신경망으로 이미지 디블렌딩 성능을 평가했다. PyTorch 기반이라 완전 자동미분과 GPU 가속이 가능해 경사 기반 최적화가 필요한 알고리즘에 활용할 수 있으며, 기존 CPU 파이프라인 대비 약 $\mathcal{O}(10^3)$의 속도 향상을 달성했다고 밝혔다.

arXiv astro-ph.GA (은하)

Breathtaking view of milky way on starry sky with bright glowing lights in dark space against black background in universe
천문·천체물리

AutoClassMK: 순수 파이썬으로 구현한 항성 2차원 MK 분광분류 신경망 공개

연구진은 파이썬과 Numpy만으로 작성한 5층 완전연결 이중 헤드 신경망 AutoClassMK를 공개했다. 이 신경망은 libr18 MK 아틀라스를 따르는 정상 항성 스펙트럼을 2차원 MK 분류 체계로 분류하며, 특수 라이브러리를 호출하지 않는 투명한 기본 코드로 구성된 점이 특징이다. 학습을 위해 libr18 및 libr18_27 MK 아틀라스를 증강해 크고 잡음이 많은 인공 학습·시험 데이터셋을 만들고 광도 분류를 단순화했으며, libr18_225 MK 아틀라스 스펙트럼의 잡음 증강본으로 예측 성능을 시험했다. 동일한 구조를 PyTorch로도 구현해 CUDA GPU에서 실행할 수 있도록 했고, 모든 코드와 데이터셋은 OpenStars 웹사이트(www.ap.smu.ca/OpenStars)에서 공개된다.

arXiv astro-ph.IM (관측기기)