Polynomial Chaos Expansion for Operator Learning

#### Контекст Операторное обучение (Operator Learning, OL) является мощным инструментом в области scientific machine learning (SciML), позволяющим оценивать отображения между пространствами функций бесконечной размерности. Одной из основных задач в этой области является обучение оператора решения уравнений в частных производных (PDE). Несмотря на то, что большинство успехов в данной области были достигнуты за счет глубоких нейронных сетей, таких как Deep Operator Networks (DeepONet) и Fourier Neural Operator (FNO), в последнее время ростую признательность набирают методы традиционной машинного обучения. Целью настоящей работы является введение метода полиномиальной хаосной разложения (Polynomial Chaos Expansion, PCE) для решения задач OL. PCE широко используется в области неопределенности (Uncertainty Quantification, UQ) и недавно привлек внимание в целях SciML. #### Метод Мы предлагаем математическую модель, позволяющую использовать PCE для аппроксимации операторов в обоих наборах условий: только на основе данных и при физической осведомленности. Модель преобразует задачу обучения оператора в решение системы уравнений для коэффициентов PCE. Это позволяет не только решать задачи операторного обучения, но и осуществлять неопределенность оценки (UQ) через простую построение процесса распределения коэффициентов. Благодаря этому, метод PCE значительно экономит ресурсы и упрощает процесс применения в прикладных задачах. #### Результаты Мы проверили процедуру на нескольких примерах PDE, включая Laplace, Poisson и Navier-Stokes уравнения. Результаты показывают, что PCE обеспечивает высокую точность оценок оператора и выполняет UQ с точностью, сопоставимой с DeepONet и FNO, но с значительно меньшими затратами по ресурсам. Наши результаты подтверждают, что PCE является эффективным и надежным методом для решения задач OL и UQ в различных прикладных сценариях. #### Значимость Метод PCE может быть применен в широком круге задач, включая моделирование систем с неопределенностью, синтез решений PDE и другие задачи SciML, где необходимо высокий уровень точности и компактности алгоритмов. Его преимущества заключаются в двусторонней эффективности: как для решения PDE, так и для построения моделей UQ. Будущие исследования будут направлены на расширение применений PCE к более сложным PDE-моделям и интеграцию с другими методами машинного обучения. #### Выводы Мы представляем полиномиальную хаосную разложение как эффективный метод для операторного обучения и неопределенности оценки в SciML. Наши результаты показывают, что PCE обеспечивает высокую точность и эффективность, делая его привлекательным вариантом для решения PDE и моделирования неопределенности. Будущие работы будут ориентированы на расширение PCE для более сложных