4D-PreNet: A Unified Preprocessing Framework for 4D-STEM Data Analysis

2508.03775v1 cs.CV, cond-mat.mtrl-sci, cs.AI, I.2.10; I.5.1; J.2 2025-08-09
Авторы:

Mingyu Liu, Zian Mao, Zhu Liu, Haoran Zhang, Jintao Guo, Xiaoya He, Xi Huang, Shufen Chu, Chun Cheng, Jun Ding, Yujun Xie

Резюме на русском

## КОНТЕКСТ И ПРОБЛЕМАТИКА В настоящее время 4D-STEM (четырехмерная сканирующая трансмиссионная электронная микроскопия) является мощным инструментом для изучения материалов на атомном и нанометровом уровнях. Этот метод предоставляет возможность получать высокоточные данные о структурной и электронной структуре материалов. Однако, несмотря на широкие возможности, 4D-STEM сталкивается с серьезными проблемами, связанными с обработкой данных. Данные, полученные при 4D-STEM, часто содержат шум, погрешности в центральном положении луча, а также искажения в виде эллиптических деформаций. Эти артефакты могут существенно исказить результаты анализа, особенно при высокопропускной обработке данных. Традиционные методы коррекции, такие как алгоритмы для устранения шума или исправления положения луча, часто оказываются специфичными для конкретных материалов и условий измерения. Это ограничивает их применимость в реальных экспериментальных условиях, где могут встречаться разнообразные типы шума, погрешностей и искажений. Более того, многие существующие методы не обеспечивают достаточно высокого уровня точности, что может привести к систематическим ошибкам в квантитативных измерениях. При этом автоматизация экспериментов и в реальном времени анализ данных требует более надежного и универсального подхода к предварительной обработке данных. В этой связи возникает потребность в разработке интегрированного и гибкого метода, который мог бы эффективно справляться с различными типами помех и искажений в данных 4D-STEM. ## ПРЕДЛОЖЕННЫЙ МЕТОД В данной работе авторы предлагают 4D-PreNet – унифицированный предварительный обработчик для 4D-STEM данных, основанный на глубоком обучении. 4D-PreNet состоит из двух основных компонентов: attention-enhanced U-Net и ResNet. Эти модели объединяются в единую конвейерную систему, которая одновременно выполняет три ключевых задачи: деноизирование (устранение шума), коррекцию положения луча (center detection) и калибровку эллиптических искажений (elliptical distortion calibration). Attention-enhanced U-Net используется для деноизирования и восстановления дифракционных паттернов, которые могут быть искажены шумом. ResNet, в свою очередь, применяется для точного определения положения луча и коррекции эллиптических деформаций. Обучение модели осуществляется на больших симуляционных наборах данных, которые включают широкий спектр шумов, погрешностей и типов искажений. Это позволяет модели хорошо адаптироваться к различным условиям эксперимента и обеспечивает высокую степень универсальности. Ключевым моментом является то, что 4D-PreNet обеспечивает конечно-структурированный подход к предварительной обработке данных, что позволяет автоматизировать процесс и улучшить качество результатов. Благодаря использованию глубоких нейронных сетей, модель может обрабатывать большие объемы данных быстро и эффективно, что критически важно для высокопропускной обработки данных в 4D-STEM. ## ЭКСПЕРИМЕНТАЛЬНЫЕ РЕЗУЛЬТАТЫ Для оценки эффективности 4D-PreNet авторы провели ряд экспериментов, используя как симуляционные, так и реальные данные. На симуляционных данных были исследованы различные уровни шума, погрешностей и искажений, чтобы оценить способность модели к обобщению. Результаты показали, что 4D-PreNet способна уменьшить среднеквадратичную ошибку (MSE) в процессе деноизирования до 50% по сравнению с традиционными методами. Также, модель достигает высокой точности в задаче определения положения луча, с ошибкой менее 0,04 пикселя. Кроме того, были проведены сравнения с традиционными алгоритмами коррекции. Результаты экспериментов показали, что 4D-PreNet не только эффективнее в устранении шума, но также лучше восстанавливает дифракционные паттерны, что важно для получения точных квантитативных измерений. Эти результаты демонстрируют преимущества интегрированного подхода, основанного на глубоком обучений, по сравнению с традиционными методами. ## ПРАКТИЧЕСКАЯ ЗНАЧИМОСТЬ 4D-PreNet имеет значительный потенциал для практического применения в области материаловедения и нанотехнологий. Благодаря его способности эффективно обрабатывать данные 4D-STEM, этот метод может стать ключевым инструментом для автоматизации экспериментов и в реальном времени анализа данных. Улучшение качества данных путем устранения шума и исправления искажений позволяет повысить точность измерений, что важно для многих приложений, включая изучение фазовых переходов, дефектов в кристаллах, и других структурных характеристик материалов. Преимущества 4D-PreNet также состоят в его универсальности и адаптивности к различным условиям эксперимента. Это позволяет использовать этот метод в различных контекстах, независимо от конкретных материалов или условий измерения. Такой подход может существенно ускорить процесс анализа и повысить надежность результатов в автоматизированных экспериментах. ## ВЫВОДЫ И ПЕРСПЕКТИВЫ В данной работе был представлен 4D-PreNet – унифицированный фреймворк для предварительной обработки данных 4D-STEM, который обеспечивает высокую точность в устранении шума, коррекции положения луча и исправлении эллиптических искажений. Результаты экспериментов показали значительные улучшения в сравнении с традиционными методами, что демонстрирует потенциал этого метода для решения ключевых проблем в области 4D-STEM анализа. В будущем можно ожидать дальнейшего развития этого подхода, включая улучшение архитектуры модели, расширение обучающих данных и интеграцию с другими методами анализа. Также, возможно расширение применения 4D-PreNet на другие типы электронной микроскопии и другие области, где критически важно высокоточное обработка данных в реальном времени.

Abstract

Automated experimentation with real time data analysis in scanning transmission electron microscopy (STEM) often require end-to-end framework. The four-dimensional scanning transmission electron microscopy (4D-STEM) with high-throughput data acquisition has been constrained by the critical bottleneck results from data preprocessing. Pervasive noise, beam center drift, and elliptical distortions during high-throughput acquisition inevitably corrupt diffraction patterns, systematically biasing quantitative measurements. Yet, conventional correction algorithms are often material-specific and fail to provide a robust, generalizable solution. In this work, we present 4D-PreNet, an end-to-end deep-learning pipeline that integrates attention-enhanced U-Net and ResNet architectures to simultaneously perform denoising, center correction, and elliptical distortion calibration. The network is trained on large, simulated datasets encompassing a wide range of noise levels, drift magnitudes, and distortion types, enabling it to generalize effectively to experimental data acquired under varying conditions. Quantitative evaluations demonstrate that our pipeline reduces mean squared error by up to 50% during denoising and achieves sub-pixel center localization in the center detection task, with average errors below 0.04 pixels. The outputs are bench-marked against traditional algorithms, highlighting improvements in both noise suppression and restoration of diffraction patterns, thereby facilitating high-throughput, reliable 4D-STEM real-time analysis for automated characterization.

Ссылки и действия