Colloidoscope: Detecting Dense Colloids in 3d with Deep Learning

Abdelwahab Kawafi,Lars Kürten,Levke Ortlieb,Yushi Yang,Abraham Mauleon Amieva,James E. Hallett,C.Patrick Royall
DOI: https://doi.org/10.48550/arXiv.2409.04603
2024-09-07
Abstract:Colloidoscope is a deep learning pipeline employing a 3D residual Unet architecture, designed to enhance the tracking of dense colloidal suspensions through confocal microscopy. This methodology uses a simulated training dataset that reflects a wide array of real-world imaging conditions, specifically targeting high colloid volume fraction and low-contrast scenarios where traditional detection methods struggle. Central to our approach is the use of experimental signal-to-noise ratio (SNR), contrast-to-noise ratio (CNR), and point-spread-functions (PSFs) to accurately quantify and simulate the experimental data. Our findings reveal that Colloidoscope achieves superior recall in particle detection (finds more particles) compared to conventional heuristic methods. Simultaneously, high precision is maintained (high fraction of true positives.) The model demonstrates a notable robustness to photobleached samples, thereby prolonging the imaging time and number of frames than may be acquired. Furthermore, Colloidoscope maintains small scale resolution sufficient to classify local structural motifs. Evaluated across both simulated and experimental datasets, Colloidoscope brings the advancements in computer vision offered by deep learning to particle tracking at high volume fractions. We offer a promising tool for researchers in the soft matter community, this model is deployed and available to use pretrained: <a class="link-external link-https" href="https://github.com/wahabk/colloidoscope" rel="external noopener nofollow">this https URL</a>.
Soft Condensed Matter,Materials Science,Statistical Mechanics
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: under the conditions of high volume fraction and low contrast, it is difficult to effectively track particles in dense colloidal suspensions using traditional methods. Specifically, traditional detection methods perform poorly when dealing with small particles or high - concentration suspensions, and are prone to problems such as missed detections (low recall rate), false detections (low precision), and coordinate detection errors. To overcome these problems, the paper proposes a new deep - learning - based method - Colloidoscope, which utilizes a 3D residual U - net architecture to enhance the tracking ability of dense colloidal suspensions through confocal microscopy. The following are the main contributions of this method: 1. **Improved detection performance**: Colloidoscope can detect more particles more accurately (high recall rate) than traditional heuristic methods while maintaining high precision (high true positive rate). This makes it perform well in handling high - volume - fraction and low - contrast scenarios. 2. **Simulated training data set**: To address the lack of precise annotations in experimental data, the researchers generated a simulated training data set that includes multiple real - world imaging conditions. These conditions include signal - to - noise ratio (SNR), contrast - to - noise ratio (CNR), and point - spread function (PSF) to ensure that the model can adapt to different experimental environments. 3. **Robustness and resolution**: Colloidoscope is highly robust to photobleached samples, which can extend the imaging time and obtain more frames. In addition, this model can also maintain sufficient small - scale resolution for classifying local structural motifs. 4. **Broad application prospects**: As an open - source tool, Colloidoscope provides researchers in the field of soft matter research with a powerful new tool that can perform particle tracking at high volume fractions, thereby promoting the development of related fields. In summary, this paper aims to improve the particle - tracking effect in high - concentration colloidal systems by introducing advanced computer vision techniques, especially deep learning, and to provide more reliable solutions for soft matter scientific research. ### Formula summary - **Signal - to - noise ratio (SNR)**: \[ SNR=\frac{\bar{x}}{s} \] where \(\bar{x}\) is the sample average brightness and \(s\) is the standard deviation. - **Contrast - to - noise ratio (CNR)**: \[ CNR = \frac{|f\bar{x}-b\bar{x}|}{b_s} \] where \(f\bar{x}\) is the foreground average brightness, \(b\bar{x}\) is the background average brightness, and \(b_s\) is the background standard deviation. - **Distance matrix \(D_{ij}\)**: \[ D_{ij}=|p_i - \hat{p}_j| \] where \(p_i\) is the true position and \(\hat{p}_j\) is the predicted position. - **Mismatch matrix \(M_{ij}\)**: \[ M_{ij}=\begin{cases} 1 & \text{if } D_{ij}>\tau\\ 0 & \text{if } D_{ij}\leq\tau \end{cases} \] - **Precision**: \[ Precision=\frac{1}{\hat{N}}\sum_{j = 1}^{\hat{N}}\left(1-\prod_{i = 1}^{N}M_{ij}\right) \] - **Recall**: \[ Recall=\frac{1}{N}\sum_{i = 1}^{N}\left(1-\prod_{j = 1}^{\hat{N}}M_{ij}\right) \] - **Average precision (AP)**: \[ AP=\frac{1}{T}\sum_{t = 0}^{T - 1}AP_t \]