3754
A GPU-accelerated Extended Phase Graph Algorithm for differentiable optimization and learning
Somnath Rakshit1, Ke Wang2, and Jonathan I Tamir3,4,5
1School of Information, The University of Texas at Austin, Austin, TX, United States, 2Electrical Engineering and Computer Sciences, University of California, Berkeley, Berkeley, CA, United States, 3Electrical and Computer Engineering, The University of Texas at Austin, Austin, TX, United States, 4Diagnostic Medicine, Dell Medical School, The University of Texas at Austin, Austin, TX, United States, 5Oden Institute for Computational Engineering and Sciences, The University of Texas at Austin, Austin, TX, United States
We present an open source, GPU-accelerated EPG simulator in PyTorch. Since the simulator is fully differentiable by means of automatic differentiation, it can be used to take derivatives with respect to sequence parameters, e.g. flip angles, as well as tissue parameters, e.g. T1 and T2.  
Figure 1: (a) Organization of the three components of the EPG algorithm, (b) 100 signals simulated using various values of T1 and T2 for a multi-echo spin-echo sequence with 60-degree refocusing flip angles.
Figure 5: The simulator is combined with a fully connected neural network for estimating T1 and T2, where the flip angle train is optimized via auto-differentiation to minimize fitting error.