Parallel Jacobian Computation Based on Distance-2 Algorithm and Its Application in Coupling Problems

The fission nuclear reactor is a typical complicated multi-physics coupling system because of the nonlinear coupled terms among different physical fields. The Newton-Krylov method is an effective method for solving the multi-physics coupling problem, featuring strong stability and a high-order conve...

Full description

Saved in:
Bibliographic Details
Main Author: LIU Lixun, ZHANG Han, PENG Xinru, DOU Qinrong, WU Yingjie, GUO Jiong, LI Fu
Format: Article
Language:English
Published: Editorial Board of Atomic Energy Science and Technology 2024-06-01
Series:Yuanzineng kexue jishu
Subjects:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1850248943893479424
author LIU Lixun, ZHANG Han, PENG Xinru, DOU Qinrong, WU Yingjie, GUO Jiong, LI Fu
author_facet LIU Lixun, ZHANG Han, PENG Xinru, DOU Qinrong, WU Yingjie, GUO Jiong, LI Fu
author_sort LIU Lixun, ZHANG Han, PENG Xinru, DOU Qinrong, WU Yingjie, GUO Jiong, LI Fu
collection DOAJ
description The fission nuclear reactor is a typical complicated multi-physics coupling system because of the nonlinear coupled terms among different physical fields. The Newton-Krylov method is an effective method for solving the multi-physics coupling problem, featuring strong stability and a high-order convergence rate. Recently, the Newton-Krylov method with an explicit Jacobian matrix has become popular. Compared with the JFNK (Jacobian-free Newton-Krylov) method which doesn’t form the Jacobian matrix explicitly, it has a better preconditioner matrix (the Jacobian matrix itself) and can achieve a more stable and fast convergence. How to calculate the Jacobian matrix efficiently and automatically is a major challenge. The finite difference method is an effective way to compute the Jacobian matrix automatically and can avoid the derivation of matrix element expressions. Besides, the serial graph coloring algorithm has been utilized to achieve an efficient computation of Jacobian. By exploiting the sparsity of the Jacobian matrix, all structurally orthogonal columns can be computed simultaneously through one function evaluation. Thus, the Jacobian computational cost can be reduced by one order of magnitude. However, when solving the larger scale problem under a distributed-memory parallel environment, the parallel graph coloring algorithm is required because the Jacobian is distributed among all processors. In this study, a Distance-2 graph coloring algorithm, arising from the field of graph theory, is applied to color the Jacobian matrix in parallel. This algorithm is performed iteratively, with each iteration consisting of four stages. Local coloring: Each processor tentatively colors its diagonal submatrix using greedy coloring algorithms; Color transfer: The colors of the diagonal submatrix are transferred to other processors to update the off-diagonal submatrix. Conflict detection: Each processor concurrently checks whether the color of the off-diagonal submatrix conflicts with the diagonal submatrix. The conflicts are gathered back to the corresponding diagonal submatrix. Color reset: Reset columns with color conflicts, waiting for recoloring in the next iteration. The entire iteration process continues until each column is correctly colored. This parallel coloring algorithm was verified by solving the steady and transient neutronic/thermal-hydraulics coupling problem arising from a simplified pressurized water reactor model. The Jacobian coloring and Jacobian computing based on this method were tested under a different number of processors. The results indicate that the Jacobian matrix computed by this parallel coloring algorithm is completely correct. The parallel coloring number only slightly increases with the number of processors. The parallel Newton-Krylov method with explicit Jacobian is more efficient and stable than the parallel JFNK method. In a word, this parallel coloring algorithm is successfully applied to the parallel Newton-Krylov method for solving the large-scale neutronic/thermal-hydraulics coupling problem.
format Article
id doaj-art-bbab0b694de04d98acfa652aabd45d6a
institution OA Journals
issn 1000-6931
language English
publishDate 2024-06-01
publisher Editorial Board of Atomic Energy Science and Technology
record_format Article
series Yuanzineng kexue jishu
spelling doaj-art-bbab0b694de04d98acfa652aabd45d6a2025-08-20T01:58:36ZengEditorial Board of Atomic Energy Science and TechnologyYuanzineng kexue jishu1000-69312024-06-015861201120910.7538/yzk.2023.youxian.0763Parallel Jacobian Computation Based on Distance-2 Algorithm and Its Application in Coupling ProblemsLIU Lixun, ZHANG Han, PENG Xinru, DOU Qinrong, WU Yingjie, GUO Jiong, LI Fu0Institute of Nuclear and New Energy Technology, Key Laboratory of Advanced Reactor Engineering and Safety of Ministry of Education, Tsinghua University, Beijing 100084, ChinaThe fission nuclear reactor is a typical complicated multi-physics coupling system because of the nonlinear coupled terms among different physical fields. The Newton-Krylov method is an effective method for solving the multi-physics coupling problem, featuring strong stability and a high-order convergence rate. Recently, the Newton-Krylov method with an explicit Jacobian matrix has become popular. Compared with the JFNK (Jacobian-free Newton-Krylov) method which doesn’t form the Jacobian matrix explicitly, it has a better preconditioner matrix (the Jacobian matrix itself) and can achieve a more stable and fast convergence. How to calculate the Jacobian matrix efficiently and automatically is a major challenge. The finite difference method is an effective way to compute the Jacobian matrix automatically and can avoid the derivation of matrix element expressions. Besides, the serial graph coloring algorithm has been utilized to achieve an efficient computation of Jacobian. By exploiting the sparsity of the Jacobian matrix, all structurally orthogonal columns can be computed simultaneously through one function evaluation. Thus, the Jacobian computational cost can be reduced by one order of magnitude. However, when solving the larger scale problem under a distributed-memory parallel environment, the parallel graph coloring algorithm is required because the Jacobian is distributed among all processors. In this study, a Distance-2 graph coloring algorithm, arising from the field of graph theory, is applied to color the Jacobian matrix in parallel. This algorithm is performed iteratively, with each iteration consisting of four stages. Local coloring: Each processor tentatively colors its diagonal submatrix using greedy coloring algorithms; Color transfer: The colors of the diagonal submatrix are transferred to other processors to update the off-diagonal submatrix. Conflict detection: Each processor concurrently checks whether the color of the off-diagonal submatrix conflicts with the diagonal submatrix. The conflicts are gathered back to the corresponding diagonal submatrix. Color reset: Reset columns with color conflicts, waiting for recoloring in the next iteration. The entire iteration process continues until each column is correctly colored. This parallel coloring algorithm was verified by solving the steady and transient neutronic/thermal-hydraulics coupling problem arising from a simplified pressurized water reactor model. The Jacobian coloring and Jacobian computing based on this method were tested under a different number of processors. The results indicate that the Jacobian matrix computed by this parallel coloring algorithm is completely correct. The parallel coloring number only slightly increases with the number of processors. The parallel Newton-Krylov method with explicit Jacobian is more efficient and stable than the parallel JFNK method. In a word, this parallel coloring algorithm is successfully applied to the parallel Newton-Krylov method for solving the large-scale neutronic/thermal-hydraulics coupling problem.newton-krylov methodsparse jacobian matrixgraph coloringfinite differencedistributed-memory parallel computing
spellingShingle LIU Lixun, ZHANG Han, PENG Xinru, DOU Qinrong, WU Yingjie, GUO Jiong, LI Fu
Parallel Jacobian Computation Based on Distance-2 Algorithm and Its Application in Coupling Problems
Yuanzineng kexue jishu
newton-krylov method
sparse jacobian matrix
graph coloring
finite difference
distributed-memory parallel computing
title Parallel Jacobian Computation Based on Distance-2 Algorithm and Its Application in Coupling Problems
title_full Parallel Jacobian Computation Based on Distance-2 Algorithm and Its Application in Coupling Problems
title_fullStr Parallel Jacobian Computation Based on Distance-2 Algorithm and Its Application in Coupling Problems
title_full_unstemmed Parallel Jacobian Computation Based on Distance-2 Algorithm and Its Application in Coupling Problems
title_short Parallel Jacobian Computation Based on Distance-2 Algorithm and Its Application in Coupling Problems
title_sort parallel jacobian computation based on distance 2 algorithm and its application in coupling problems
topic newton-krylov method
sparse jacobian matrix
graph coloring
finite difference
distributed-memory parallel computing
work_keys_str_mv AT liulixunzhanghanpengxinrudouqinrongwuyingjieguojionglifu paralleljacobiancomputationbasedondistance2algorithmanditsapplicationincouplingproblems