Scholarlink

<p>Photometric classification of Type Ia supernovae (SNe Ia) is critical for cosmological studies but remains difficult due to class imbalance and observational noise. While deep learning models have been explored, they are often resource-intensive and lack interpretability. We present a computationally efficient and interpretable classification framework that maintains high performance on imbalanced datasets. We emphasize the use of PR-AUC and F1-score as more informative metrics than ROC-AUC in severely imbalanced settings. Using an XGBoost ensemble optimized via Bayesian hyperparameter tuning, we classified light curves from the Supernova Photometric Classification Challenge (SPCC), comprising 21,318 events with a 3.19 imbalance ratio (non-Ia to Ia). Our model achieved a PR-AUC of <InlineEquation ID="IEq1"> <EquationSource Format="TEX">\(0.993^{+0.03}_{-0.02}\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>0</mn> <mo>.</mo> <msubsup> <mn>993</mn> <mrow> <mo>-</mo> <mn>0.02</mn> </mrow> <mrow> <mo>+</mo> <mn>0.03</mn> </mrow> </msubsup> </mrow> </math></EquationSource> </InlineEquation>, an F1-score of <InlineEquation ID="IEq2"> <EquationSource Format="TEX">\(0.923 \pm 0.008\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>0.923</mn> <mo>±</mo> <mn>0.008</mn> </mrow> </math></EquationSource> </InlineEquation>, and a ROC-AUC of <InlineEquation ID="IEq3"> <EquationSource Format="TEX">\(0.976 \pm 0.004\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>0.976</mn> <mo>±</mo> <mn>0.004</mn> </mrow> </math></EquationSource> </InlineEquation>, matching or exceeding deep learning performance on precision-recall trade-offs, while using fewer resources. Despite slightly lower overall accuracy, our method balances false positives and false negatives, improving the efficiency of spectroscopic follow-up. We show that optimized ensemble models offer a reproducible and lightweight alternative to complex architectures, particularly for large-scale surveys, such as the Legacy Survey of Space and Time (LSST), where transparency and efficiency are essential.</p>

错误：搜索内容不能为空，请输入英文关键词
错误：关键词超出字数限制，请精简
高级检索

Optimizing supernova classification with interpretable machine learning models

  • 期刊论文

  • 发表日期：

    2025年11月26日

查看全文
  • Anurag Garg

摘要

Photometric classification of Type Ia supernovae (SNe Ia) is critical for cosmological studies but remains difficult due to class imbalance and observational noise. While deep learning models have been explored, they are often resource-intensive and lack interpretability. We present a computationally efficient and interpretable classification framework that maintains high performance on imbalanced datasets. We emphasize the use of PR-AUC and F1-score as more informative metrics than ROC-AUC in severely imbalanced settings. Using an XGBoost ensemble optimized via Bayesian hyperparameter tuning, we classified light curves from the Supernova Photometric Classification Challenge (SPCC), comprising 21,318 events with a 3.19 imbalance ratio (non-Ia to Ia). Our model achieved a PR-AUC of \(0.993^{+0.03}_{-0.02}\) 0 . 993 - 0.02 + 0.03 , an F1-score of \(0.923 \pm 0.008\) 0.923 ± 0.008 , and a ROC-AUC of \(0.976 \pm 0.004\) 0.976 ± 0.004 , matching or exceeding deep learning performance on precision-recall trade-offs, while using fewer resources. Despite slightly lower overall accuracy, our method balances false positives and false negatives, improving the efficiency of spectroscopic follow-up. We show that optimized ensemble models offer a reproducible and lightweight alternative to complex architectures, particularly for large-scale surveys, such as the Legacy Survey of Space and Time (LSST), where transparency and efficiency are essential.