Classifying the rotation of bacteria using neural networks

Sammanfattning: Bacteria can quickly spread throughout the human body, making certain diseases hard or impossible to cure. In order to understand how the bacteria can initiate and develop into an infection, microfluidic chambers in a lab environment are used as a template of how bacteria reacts to different types of flows. However, accurately tracking the movement of bacteria is a difficult task, where small objects has to be captured with a high resolution and digitally analysed with computationally heavy methods. Popular imaging methods utilise digital holographic microscopy, where three-dimensional movement is captured in two-dimensional images by numerical reconstruction of the diffraction of light. Since numerical reconstructions become computationally heavy when a good accuracy is required, this master's thesis work focus on evaluating the possibility of using convolutional neural networks to quickly and accurately determine the spatial properties of bacteria. By thorough testing and analysis of state of the art and old networks a new network design is presented, designed to eliminate as many imaging issues as possible. We found that there are certain network design choices that help with reducing the overall error of the system, and with a well chosen training set with sensible augmentations, some networks were able to reach a 60% classification accuracy when determining the vertical rotation of the bacteria. Unfortunately, due to the lack of experimental data where the ground-truth is known, not much experimental testing could be performed. However, a few tests showed that images of high quality could be classified within the expected range of vertical rotation.