Код BeAR на GPU разбирает атмосферу WASP-77Ab: 17 параметров за 14 минут на восьми RTX 4090
Авторы бернского кода BeAR сообщили о ретривале спектров экзопланет на GPU: одна модель занимает около 1 мс, результаты для WASP-77Ab совпадают с опубликованными.
Авторы Bern Atmospheric Retrieval code (BeAR), преемника Helios-r2, представили основанный на GPU пакет для восстановления атмосфер экзопланет по спектрам высокого, низкого разрешения и по их совместному набору. Работа опубликована как arXiv · препринт и ещё не прошла рецензирование. Код открыт и доступен на GitHub arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗.
Проверку провели на дневных спектрах горячего юпитера WASP-77Ab, полученных на Gemini South/IGRINS и JWST/NIRSpec arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗.
Как устроен пакет
BeAR выполняет совместный ретривал, складывая логарифмы правдоподобия данных высокого и низкого разрешения, рассчитанные по одной модельной атмосфере arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗. Для высокого разрешения реализованы известные кросс-корреляционные правдоподобия, где масштаб глубины линий либо подбирается аналитически, либо сэмплируется как свободный параметр. Есть также их попиксельное обобщение, быстрая фильтрация моделей и повторная инжекция модели, согласованная с этой фильтрацией
arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗.
Прямая модель и правдоподобие целиком считаются на графическом процессоре. Одна прямая модель занимает около 1 мс на NVIDIA GH200, что, по словам авторов, даёт миллионы моделей в минуту на GPU-кластере arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗. Для сравнения: прежний ретривал данных IGRINS для WASP-77Ab занимал около трёх суток на 24 ядрах CPU и одном GPU
arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗.
Результаты для WASP-77Ab
В раздельных ретривалах спектров IGRINS и NIRSpec BeAR воспроизводит опубликованные содержания, отношения C/O и металличности в пределах 1,5 сигма. Предпочтение 13CO, тепловая структура и контраст линий совпадают в пределах 1 сигма arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗. Совместный ретривал подтверждает субсолнечную металличность и околосолнечное отношение C/O
arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗.
Полный ретривал IGRINS с 17 свободными параметрами занимает от 27 до 42 минут на одной NVIDIA GeForce RTX 5080 или около 14 минут на восьми RTX 4090 arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗.
Тесты инжекции и восстановления показывают ограничение: апостериорные распределения занижают неопределённости отношений содержаний примерно вдвое arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗. Авторы отмечают, что результаты высокого разрешения зависят от выбора методики анализа, например от удаления теллурических и звёздных линий, поэтому их нужно проверять независимыми кодами. Открытых кодов для этого мало, а время счёта в часы и сутки делает такие проверки дорогими
arxiv.org
arXiv · препринтarxiv.orgОткрыть источник ↗.
