The application belongs to the field of
brain white matter fiber track prediction, and relates to a
brain white matter fiber track prediction method and
system based on a pre-trained basic model. The method adopts an MAE architecture, and performs self-supervised pre-training on large-scale brain
fiber track data through a
mask-reconstruction mechanism to learn
global topology and local geometric features of neural fibers. In the training stage, a
hybrid progressive
mask strategy is introduced. In the initial stage of the model, continuous segment masks are mainly used, and gradually transition to random point masks, so as to realize hierarchical learning of structural patterns and robust expression of features. After pre-training is completed, the model is fine-tuned, so that only the input of the two end point coordinates of the fiber can reconstruct and predict the intermediate track, and the fiber path
inference based on the end point is realized. In the fine-tuning stage, the
encoder structure is kept stable, the decoder is guided by the end point features to generate complete tracks conforming to the
structural distribution of the
brain white matter fiber, and the technical problems existing in the prior art are solved.