Explainable AI for Curie Temperature Prediction in Magnetic Materials

#### Контекст Предсказание Кюри (Curie) температуры в магнитных материалах является важной задачей в сфере материалов и физики. Температура Кюри определяет точку, при которой материал перестает быть магнитным. Изучение этого показателя необходимо для разработки материалов с управляемыми магнитными свойствами. Несмотря на развитие машинного обучения в материаловедении, оценка Curie-температуры остается недостаточно точной и понятной. Большинство существующих моделей не предоставляют понятных выводов о том, какие физико-химические характеристики влияют на Curie-температуру. Этот недостаток снижает доверие к моделям и ограничивает их применение в научных и промышленных областях. #### Метод Мы применяем машинное обучение для предсказания Curie-температур с использованием базы данных NEMAD. В качестве входных данных используются химические и физические описательные признаки материалов. Для повышения точности и понятности результатов мы модифицировали данные, добавив дополнительные компоненты, такие как свойства атомов и магнитные моменты. Мы использовали модели Extra Trees Regressor, Random Forest и Gradient Boosting. Для понимания влияния отдельных признаков на результат использовались методы SHAP. Кроме того, мы применили кластеризацию k-means для анализа химически различных групп материалов. #### Результаты Модель Extra Trees Regressor показала самый высокий R2-результат, достигая значения 0,85 ± 0,01 при кросс-валидации. Это улучшение связано с использованием дополнительных признаков и оптимизированным предварительным процессом обработки данных. Анализ SHAP позволил выявить ключевые факторы, влияющие на Curie-температуру, в том числе средний атомный номер и магнитный момент. Использование кластеризации позволило выделить химически различные группы материалов, что дало более глубокое понимание взаимосвязи между химическими свойствами и Curie-температурой. #### Значимость Результаты приложения Explainable AI могут быть использованы в промышленности и науке для оптимизации материалов с управляемыми магнитными свойствами. Использование понятных моделей обучения позволяет улучшить надежность и доверие к результатам. Этот подход может быть распространен на другие задачи, где необходимо объяснение результатов моделирования. Это дает возможность расширить применение машинного обучения в области материалов, где важно понимать причины модельных выводов. #### Выводы Мы успешно применили машинное обучение для точного предсказания Curie-температур с помощью дополнительных признаков и объяснительных методов. Наши результаты показывают, что Extra Trees Regressor является наиболее эффективным для этой задачи. Дальней