Computing

, Volume 95, Supplement 1, pp 771–784

GPU-accelererated regularisation of large diffusion-tensor volumes

Article

Abstract

We discuss the benefits, difficulties, and performance of a GPU implementation of the Chambolle–Pock algorithm for TGV (total generalised variation) denoising of medical diffusion tensor images. Whereas we have previously studied the denoising of 2D slices of \(2 \times 2\) and \(3 \times 3\) tensors, attaining satisfactory performance on a normal CPU, here we concentrate on full 3D volumes of data, where each 3D voxel consists of a symmetric \(3 \times 3\) tensor. One of the major computational bottle-necks in the Chambolle–Pock algorithm for these problems is that on each iteration at each voxel of the data set, a tensor potentially needs to be projected to the positive semi-definite cone. This in practise demands the QR algorithm, as explicit solutions are not numerically stable. For a \(128 \times 128 \times 128\) data set, for example, the count is 2 megavoxels, which lends itself to massively parallel GPU implementation. Further performance enhancements are obtained by parallelising basic arithmetic operations and differentiation. Since we use the relatively recent OpenACC standard for the GPU implementation, the article includes a study and critique of its applicability.

Keywords

DTI Regularisation Medical imaging GPU Open ACC 

Mathematics Subject Classification (2010)

92C55 94A08 26B30 49M29 

References

  1. 1.
    Bredies K, Kunisch K, Pock T (2011) Total generalized variation. SIAM J Imaging Sci 3:492–526. doi:10.1137/090769521 MathSciNetCrossRefGoogle Scholar
  2. 2.
    Valkonen T, Bredies K, Knoll F (2012) Total generalised variation in diffusion tensor imaging. SFB-report no. 2012–003, Karl-Franzens University of Graz, Graz. http://math.uni-graz.at/mobis/publications/SFB-Report-2012-003.pdf
  3. 3.
    Bredies K, Kunisch K, Valkonen T (2013) Properties of \(L^1\)-\(\text{ TGV}^2\): the one-dimensional case. J Math Anal Appl 398, 438–454. doi: 10.1016/j.jmaa.2012.08.053, http://math.uni-graz.at/mobis/publications/SFB-Report-2011-006.pdf
  4. 4.
    Knoll F, Bredies K, Pock T, Stollberger R (2011) Second order total generalized variation (TGV) for MRI. Magn Reson Med 65(2):480–491. doi:10.1002/mrm.22595 CrossRefGoogle Scholar
  5. 5.
    Basser PJ, Jones DK (2002) Diffusion-tensor MRI: theory, experimental design and data analysis— a technical review. NMR Biomed 15(7–8):456–467. doi:10.1002/nbm.783
  6. 6.
    Kingsley P (2006) Introduction to diffusion tensor imaging mathematics: parts I–III. Concepts Magn Reson Part A 28(2):101–179. doi:10.1002/cmr.a.20048; doi:10.1002/cmr.a.20049; doi:10.1002/cmr.a.20050
  7. 7.
    Tournier JD, Mori S, Leemans A (2011) Diffusion tensor imaging and beyond. Magn Reson Med 65(6):1532–1556. doi:10.1002/mrm.22924 CrossRefGoogle Scholar
  8. 8.
    Gudbjartsson H, Patz S (1995) The Rician distribution of noisy MRI data. Magn Reson Med 34(6):910–914CrossRefGoogle Scholar
  9. 9.
    Herbst M, Maclaren J, Weigel M, Korvink J, Hennig J, Zaitsev M (2011) Prospective motion correction with continuous gradient updates in diffusion weighted imaging. Magn Reson Med. doi:10.1002/mrm.23230
  10. 10.
    Aksoy M, Forman C, Straka M, Skare S, Holdsworth S, Hornegger J, Bammer R (2011) Real-time optical motion correction for diffusion tensor imaging. Magn Reson Med 66(2):366–378. doi:10.1002/mrm.22787 CrossRefGoogle Scholar
  11. 11.
    Arsigny V, Fillard P, Pennec X, Ayache N (2005) Fast and simple computations on tensors with log-euclidean metrics. Technical report 5584, INRIAGoogle Scholar
  12. 12.
    Setzer S, Steidl G, Popilka B, Burgeth B (2009) Variational methods for denoising matrix fields. In: Weickert J, Hagen H (eds) Visualization and processing of tensor fields. Springer, Berlin, pp 341–360CrossRefGoogle Scholar
  13. 13.
    Tschumperlé D, Deriche R (2001) Diffusion tensor regularization with constraints preservation. In: Proceedings of the 2001 IEEE computer society conference on computer vision and pattern recognition (CVPR), vol 1, pp 948–953Google Scholar
  14. 14.
    Chambolle A, Pock T (2011) A first-order primal–dual algorithm for convex problems with applications to imaging. J Math Imaging Vis 40:120–145. doi:10.1007/s10851-010-0251-1 MathSciNetMATHCrossRefGoogle Scholar
  15. 15.
    OpenACC-standard.org (2011) The openACC application programming interface. http://www.openacc-standard.org/
  16. 16.
    NVIDIA Corporation: CUDA parallel programming and computing platform. http://www.nvidia.com/cuda
  17. 17.
    Bishop RL, Goldberg SI (1980) Tensor analysis on manifolds. Dover Publications, New YorkGoogle Scholar
  18. 18.
    Federer H (1969) Geometric measure theory. Springer, BerlinMATHGoogle Scholar
  19. 19.
    Temam R (1985) Mathematical problems in plasticity. Gauthier-Villars, ParisGoogle Scholar
  20. 20.
    Bredies K, Valkonen T (2011) Inverse problems with second-order total generalized variation constraints. In: Proceedings of SampTA 2011, 9th international conference on sampling theory and applications, Singapore. http://tuomov.iki.fi/mathematics/SampTA2011.pdf
  21. 21.
    Valkonen T (2008) Diff-convex combinations of Euclidean distances: a search for optima. No. 99 in Jyväskylä studies in computing. PhD thesis, University of Jyväskylä. http://tuomov.iki.fi/mathematics/thesis.pdf
  22. 22.
    Chambolle A (2004) An algorithm for total variation minimization and applications. J Math Imaging Vis 20(1):89–97MathSciNetGoogle Scholar
  23. 23.
    Pock T, Chambolle A (2011) Diagonal preconditioning for first order primal-dual algorithms in convex optimization. In: Proceedings of the 2011 IEEE international conference on computer vision (ICCV), pp 1762–1769. doi: 10.1109/ICCV.2011.6126441
  24. 24.
    Freiberger M, Egger H, Liebmann M, Scharfetter H (2011) High-performance image reconstruction in fluorescence tomography on desktop computers and graphics hardware. Biomed Opt Express 2(11):3207–3222. doi:10.1364/BOE.2.003207; http://www.opticsinfobase.org/boe/abstract.cfm?URI=boe-2-11-3207
  25. 25.
    Amorim RM, Haase G, Liebmann M, dos Santos RW (2009) Comparing CUDA and OpenGL implementations for a Jacobi iteration. In: Proceedings of the (HPCS 2009), pp 22–32Google Scholar
  26. 26.
    Smari WW, McIntire JP (eds) (2009) Proceedings of the 2009 international conference on high performance computing and simulation (HPCS 2009), Leipzig, Germany, 21–24 June 2009Google Scholar

Copyright information

© Springer-Verlag Wien 2013

Authors and Affiliations

  1. 1.Institute for Mathematics and Scientific ComputingUniversity of GrazGrazAustria

Personalised recommendations