=== RESUMEN ===

Resumen Técnico: OxiGen – Generación de Cristales Consciente del Estado de Oxidación

Planteamiento del Problema

Los modelos generativos ofrecen una vía para acelerar el descubrimiento de materiales inorgánicos mediante el diseño inverso, pero generar cristales experimentalmente realizables sigue siendo un desafío significativo. Aunque los modelos generativos de cristales existentes pueden producir estructuras con neutralidad de carga, a menudo no logran reproducir las distribuciones de estados de oxidación observadas en los materiales sintetizados. Muchos cristales generados dependen de estados de oxidación raros o químicamente improbables para satisfacer la neutralidad de carga, una limitación que no siempre es capturada por métricas de estabilidad estándar como la energía sobre el casco de convexidad (energy above the convex hull). En consecuencia, estos modelos pueden generar estructuras que son teóricamente estables pero químicamente implausibles o difíciles de sintetizar. El problema central es la falta de representación y cumplimiento explícito de las frecuencias de los estados de oxidación y de la neutralidad de carga global dentro del proceso generativo.

Metodología

Los autores proponen OxiGen, un modelo de difusión de cristales que representa explícitamente los estados de oxidación durante la generación y garantiza la neutralidad de carga por construcción.

1. Representación de Especies

A diferencia de los modelos previos que generan identidades de elementos ($A$) e infieren los estados de oxidación implícitamente, OxiGen genera especies ($S$), definidas como pares elemento–estado de oxidación $(a, z)$. El cristal se representa como $M = (S, F, L)$, donde $F$ son las coordenadas fraccionarias y $L$ es la matriz de la red. El vocabulario $S$ consiste en especies observadas en materiales sintetizados. Esta representación explícita permite al modelo aprender las frecuencias de los estados de oxidación directamente de los datos de entrenamiento.

2. Inferencia Estructurada mediante un Autómata de Estados Finitos

Un desafío crítico en la generación de especies es asegurar la restricción global de que la suma de los estados de oxidación sea cero (neutralidad de carga). El muestreo independiente de las distribuciones por sitio suele violar esta restricción. Para abordar esto, OxiGen introduce una capa de salida estructurada que realiza una inferencia exacta sobre un autómata de estados finitos (FSA).

• Formulación del FSA: La restricción de neutralidad de carga se modela como un proceso aditivo donde el estado representa la carga acumulada. Las asignaciones válidas corresponden a rutas desde un estado inicial (carga 0) hasta un estado de aceptación (carga 0) tras $N$ sitios.
• Programación Dinámica: El modelo emplea un algoritmo de paso hacia adelante–atrás (adaptado de los Modelos Ocultos de Márkov) para computar las probabilidades marginales exactas y realizar la decodificación de Máximaر Posterior (MAP) bajo la restricción.
  • Paso hacia adelante (Forward Pass): Computa la masa de probabilidad de todos los prefijos que terminan con una carga acumulada específica.
  • Paso hacia atrás (Backward Pass): Computa la masa de probabilidad de los sufijos que completan una carga dada hacia cero.
• Muestreo: Durante el paso de difusión inversa, OxiGen muestrea una asignación de especies completa y con neutralidad de carga $\hat{S}_0$ a partir de la distribución estructurada $\tilde{p}_\theta(S_0 | M_t, t)$ utilizando los pesos de retroceso calculados. Esto asegura que cada cristal generado satisfaga la neutralidad de carga por construcción, sin requerir filtrado post-hoc.

3. Arquitectura del Modelo

OxiGen se basa en la arquitectura MatterGen, una red neuronal de grafos equivariante a SE(3). Retiene los procesos de difusión continuos para las coordenadas fraccionarias y las matrices de la red, pero modifica el proceso de difusión discreta para las especies. Utiliza un proceso de difusión discreta de estado absorbente (basado en MDLM) donde los sitios son enmascarados y luego denotados. La capa de salida estructurada se aplica durante el muestreo, mientras que se utiliza el proceso de difusión enmascarada factorizado estándar para el entrenamiento (aunque se exploró un objetivo de entrenamiento estructurado en estudios de ablación).

Contribuciones Clave

1. Capa de Salida Estructurada: El desarrollo de una capa que utiliza programación dinámica sobre un autómata de estados finitos para realizar la inferencia de neutralidad de carga exacta para modelos de difusión discreta.
2. Modelo OxiGen: La introducción de un modelo de difusión de cristales que genera explícitamente especies (elemento + estado de oxidación) y garantiza la neutralidad de carga global durante el proceso de muestreo.
3. Validación Empírica: Demostración de que OxiGen mejora sustancialmente la fidelidad del estado de oxidación, genera una mayor tasa de cristales estables, únicos y novedosos en comparación con las líneas base, y mantiene una alta validez composicional incluso bajo condicionamiento de propiedades.

Resultados

Los autores evaluaron OxiGen en el conjunto de datos MP-20-OS (un subconjunto de Materials Project con etiquetas de estado de oxidación) contra cinco líneas base sólidas: DiffCSP, Mattergen, FlowMM, OMatG y Crystalite.

• Fidelidad del Estado de Oxidación: OxiGen logró la mayor Puntuación de Frecuencia de Estado de Oxidación ($f_{OS}$) (90.65%) y la menor Distancia de Distribución de Estado de Oxidación ($d_{OS}$) (0.0911) entre todos los métodos. Se ajustó estrechamente a las distribuciones del conjunto de prueba de los estados de oxidación comunes, mientras que las líneas base dependían más de estados poco comunes.
• Estabilidad y Novedad: OxiGen generó la mayor tasa de cristales Estables, Únicos y Novedosos (S.U.N.) (26.38%), superando a todas las líneas base. También logró el menor RMSD de relajación (0.1679 Å), lo que indica una alta fidelidad estructural.
• Generación Condicional: En experimentos de generación condicionada por brecha de banda (band-gap), OxiGen mantuvo una validez composicional consistentemente alta a través de las brechas de banda objetivo (1–7 eV), mientras que líneas base como MatterGen experimentaron un fuerte declive en la validez a medida que los objetivos se alejaban de la distribución de entrenamiento.
• Estudios de Ablación: Eliminar la capa de salida estructurada resultó en una caída significativa en la neutralidad de carga (del 100% a ~39%) y en la fidelidad del estado de oxidación. El uso de la representación de especies por sí solo proporcionó solo mejoras modestas, resaltando la necesidad de la capa de inferencia estructurada.

Significado y Reivindicaciones

El artículo afirma que OxiGen demuestra cómo la incorporación de conocimiento químico previo —específicamente a través de heurísticas simples como la neutralidad de carga y las frecuencias de estados de oxidación observadas— puede guiar a los modelos generativos hacia materiales químicamente más plausibles. Al representar explícitamente los estados de oxidación y garantizar las restricciones mediante inferencia estructurada, el modelo cierra la brecha entre la estabilidad teórica y la realizabilidad experimental.

Los autores señalan que su enfoque requiere solo modificaciones mínimas a las arquitecturas de modelos de difusión existentes, lo que sugiere que podría integrarse fácilmente en otros modelos generativos de cristales. Posicionan a OxiGen como un paso hacia un diseño inverso más fiable, donde los candidatos generados no solo son estables, sino también químicamente válidos de acuerdo con las reglas establecidas de la química inorgánica.

Limitaciones Reconocidas:
Los autores declaran explícitamente que OxiGen hereda las limitaciones de los estados de oxidación enteros formales, que pueden ser ambiguos o indefinidos para sistemas con una covalencia sustancial, deslocalización de carga o enlaces metálicos. Además, el modelo depende de las asignaciones de estado de oxidación inferidas por su predictor, que pueden contener errores. El método actualmente garantiza la neutralidad de carga, pero aún no garantiza otras restricciones como el equilibrio de la electronegatividad o la consistencia de la coordinación local.