A Feature Extraction Framework for 3D Scientific Voxel Object Using SVD and Neural Network

Huang Ye,Zhichen Feng,Jian Zhang,Yaqian Gao
DOI: https://doi.org/10.1109/IJCNN60899.2024.10649920
2024-06-30
Abstract:Latest advances in computational methods and high-performance computing have enabled large-scale scientific simulations to become feasible. However, efficiently analyzing the resulting large datasets remains challenging. Typically, 3D voxel data occupies huge storage space which causes serious storage pressure. Performing real-time feature extraction during simulations could mitigate this demand. In this paper, We propose a three-decker structure framework for locating and extracting voxel object features such as pose, class, and size. The framework first utilizes a 3D CNN to localize the object’s center and size along each axis, enabling adaptive cropping of the target from the original voxel space. Then the singular value decomposition (SVD) will applied to the extracted object to preliminarily extract and refine its posture. Finally, a neural network will refine the SVD-extracted pose and predict the category and size. Finally, a neural network is utilized to infer residual pose corrections as well as category and size information for the object after preliminary pose alignment via SVD decomposition. Our method achieves accurate feature extraction for voxel objects with complex poses. More importantly, we achieve pose estimation without an initial position. By storing extracted features instead of full data, we significantly reduce storage needs. We validate our approach using the phase field simulations dataset and ModelNet40 dataset. At last, based on the proposed framework we develop an in-situ feature extraction library for running with large-scale scientific computing programs and performing real-time feature extraction on the large amount of computation data it generates. As a concrete running example, we selected the microstructure evolution program governed by the phase-field method for a common run and achieved high-accuracy feature extraction.
Computer Science,Engineering
What problem does this paper attempt to address?