Generative Models Applied to Inverse Problems

Lead Research Organisation: University of Bath
Department Name: Mathematical Sciences

Abstract

My research works at the intersection of deep learning and inverse problems in imaging. Solving an inverse problem is the task of computing an unknown physical quantity given indirect measurements via a (usually) known forward model. A typical example is when imaging technologies are used in medicine, engineering, astronomy and geophysics. Interesting Inverse Problems are nearly always ill-posed and addressing this is critical in applications where decision making is based on the recovered image. In particular, sometimes the data recovered is not sufficient to solve the reconstruction problem accurately and some form of prior knowledge about the object must be incorporated in the solution process. Developing new and improving existing approaches to these cases is the aim of my project.

The proposed approaches uses deep learning methods to learn a generative model. Generative models implicitly model high-dimensional distributions of data from observations. The learnt distribution can then be used as a prior when solving the inverse problem, ensuring solutions are feasible. Learnt priors could provide more specific information than that of a hand-crafted prior while remaining flexible to changes in the forward problem. For example, we could learn a generative model that can produce all feasible brain heart and lung CT images. Then receiving low radiation dose data from the CT scanner, we search through the set of feasible images to find the image that best fits the data. The resulting image will be high quality despite the small amount of provided data.

There is wide scope for novel engineering and/or physical sciences research. The resulting optimisation required for finding the image produced by the generative model that best fits the data is both a non-linear and non-convex problem and will require application or development of state of the art techniques. Numerical analysis tools are needed to bound errors and convergence. Mathematical approaches working closely with computer scientists also need to be developed for checking and ensuring that the ground truth images can be produced by the generative model. This will add to a fast growing research area including variational autoencoders and generative adversarial networks.

Planned Impact

The impact of the SAMBa CDT will occur principally through the following two pathways:

1. Direct engagement with industrial partners, leading to PhD projects that are collaborative with industry, and that are focussed on topics with direct industrial impact.

2. The production of PhD graduates with
(a) the mathematical, statistical and computational technical skill sets that have been identified as in crucial demand both by EPSRC and by our industrial partners, coupled to
(b) extensive experience of industrial collaboration.

The underlying opportunity that SAMBa provides is to train graduates to have the ability to combine complex models with 'big data'. Such people will be uniquely equipped to deliver impact: whether they continue with academic careers or move directly to posts in industry, through quantitative modelling, they will provide the information that gives UK businesses competitive advantages. Our industrial partners make it clear to us that competitiveness in the energy, manufacturing, service, retail and financial sectors is increasingly dependent on who can best and most quickly analyse the huge datasets made available by the present information revolution.

During their training as part of SAMBa, these students will have already gained experience of industrial collaboration, through their PhD projects and/or the Integrated Think Tanks (ITTs) that we propose, that will give all SAMBa students opportunities to develop these transferable skills. PhD projects that involve industrial collaboration, whether arising from ITTs or not, will themselves deliver economic and social benefits to UK through the private companies and public sector organisations with which SAMBa will collaborate.

We emphasise that Bath is at the forefront of knowledge transfer (KT) activities of the kind needed to translate our research into impact. Our KT agenda has recently been supported by KT Accounts and Impact Acceleration Accounts from EPSRC (£4.9M in total) and a current HEFCE HEIF allocation of £2.4M. Bath is at the forefront of UK activity in KTPs, having completed 150 and currently holding 16 KTP contracts worth around £2.5M.

The SAMBa ITTs are an exciting new mechanism through which we will actively look for opportunities to turn industrial links into research partnerships, supported in the design of these projects by the substantial experience available across the University.

More widely, we envisage impact stemming from a range of other activities within SAMBa:

- We will look to feed the results of projects involving ecological or epidemiological data directly into environmental and public health policy. We have done this successfully many times and have three REF Case Studies describing work of this nature.

- Students will be encouraged to make statistical tools available as open source software. This will promote dissemination of their research results, particularly beyond academia. There is plenty of recent evidence that such packages are taken up and used.

- Students will discuss how to use new media to promote the public understanding of science, for example contributing to projects such as Wikipedia.

- Students will be encouraged to engage in at least one outreach activity. Bath is well known for its varied, and EPSRC-supported, public engagement activities that include Royal Institution Masterclasses, coaching the UK Mathematics Olympiad team, and reaching 50 000 people in ten days with an exhibit at the Royal Society's 350th Anniversary Summer Exhibition in 2010.

Publications

10 25 50
 
Description My research looks at the intersection of deep learning and inverse problems in imaging. Solving an inverse problem is the task of computing an unknown physical quantity given indirect measurements via a (usually) known forward model. A typical example is when imaging technologies are used in medicine, engineering, astronomy and geophysics. Interesting Inverse Problems are nearly always ill-posed and addressing this is critical in applications where decision-making is based on the recovered image. In particular, sometimes the data recovered is not sufficient to solve the reconstruction problem accurately and some form of prior knowledge about the object must be incorporated into the solution process.

This project too three avenues for development in this area:
1) We investigated and compared approaches for incorporating generative model deep learning approaches into a variational framework for inverse problems. Generative models implicitly model high-dimensional distributions of data from observations. The learned distribution can then be used as a prior when solving the inverse problem, ensuring solutions are feasible. Learnt priors could provide more specific information than that of a hand-crafted prior while remaining flexible to changes in the forward problem. For example, we could learn a generative model that can produce all feasible knee MRI images. Then receiving undersampled data from the MRI scanner, we search through the set of feasible images to find the image that best fits the data. The resulting image will be high quality despite the small amount of provided data. We considered cases where the reconstructed images were restricted to being in the range of the generative model and also close to the range of the generative model. In addition, essential criteria for generative models applied to inverse problems were highlighted with the aim of guiding future research.

2) In the second project we further investigated the concept of "close to" in restricting reconstructed images to be close to the range of the generative model. We were able to use a VAE with structured image covariance to learn a weighted distance metric that was able to provide data-driven criteria to assess the quality of reconstructed images.

3) In this final project, we consider learning generative models for use in inverse problems without large amounts of high-quality ground truth images.
Exploitation Route The scientific knowledge provided in presentations, pre-prints and the thesis will be of use to the inverse problems community in the fast-growing area of deep learning and inverse problems.
Sectors Digital/Communication/Information Technologies (including Software),Healthcare

 
Description Doctoral College Placement Support Fund
Amount £1,350 (GBP)
Organisation University of Bath 
Sector Academic/University
Country United Kingdom
Start 08/2019 
End 11/2019
 
Description SIAM Student Travel Award
Amount $50 (USD)
Organisation Society for Industrial and Applied Mathematics 
Sector Charity/Non Profit
Country United States
Start 03/2022 
End 03/2022
 
Description Santander Scholarships Research | Santander Mobility Award
Amount £1,150 (GBP)
Organisation Santander Universities 
Sector Private
Country United Kingdom
Start 02/2022 
End 05/2022
 
Description IS 
Organisation University of Sussex
Country United Kingdom 
Sector Academic/University 
PI Contribution Expertise, time and computing facilities.
Collaborator Contribution Collaborators provided expertise and advice
Impact Academic collaboration on a project "Compressed sensing MRI using VAEs with structured image covariance"
Start Year 2020
 
Description STEM for Britain 
Form Of Engagement Activity Participation in an activity, workshop or similar
Part Of Official Scheme? No
Geographic Reach National
Primary Audience Policymakers/politicians
Results and Impact I was selected to present a poster based on my research at the 2020 STEM for Britain event. I presented a poster entitled 'Solving inverse imaging problems using generative machine learning methods'. The event is designed for early career researchers to present and discuss "ground-breaking" and frontier UK research and R&D to Members of both Houses of Parliament at Westminster with the aim of fostering greater dialogue and engagement between early-stage researchers and Members both in Westminster and in their Constituencies. The event raises the profile of Britain's early-stage researchers at Parliament and elsewhere and contributes to various national initiatives e.g. the UK's British Science Week. In addition, it was an opportunity for personal interaction between all researchers and gain an awareness of the challenges and excitement in other areas of research.
Year(s) Of Engagement Activity 2020
URL https://www.bath.ac.uk/announcements/bath-mathematician-takes-her-research-to-parliament/