-
ViM-UNet: Vision Mamba for Biomedical Segmentation
Authors:
Anwai Archit,
Constantin Pape
Abstract:
CNNs, most notably the UNet, are the default architecture for biomedical segmentation. Transformer-based approaches, such as UNETR, have been proposed to replace them, benefiting from a global field of view, but suffering from larger runtimes and higher parameter counts. The recent Vision Mamba architecture offers a compelling alternative to transformers, also providing a global field of view, but…
▽ More
CNNs, most notably the UNet, are the default architecture for biomedical segmentation. Transformer-based approaches, such as UNETR, have been proposed to replace them, benefiting from a global field of view, but suffering from larger runtimes and higher parameter counts. The recent Vision Mamba architecture offers a compelling alternative to transformers, also providing a global field of view, but at higher efficiency. Here, we introduce ViM-UNet, a novel segmentation architecture based on it and compare it to UNet and UNETR for two challenging microscopy instance segmentation tasks. We find that it performs similarly or better than UNet, depending on the task, and outperforms UNETR while being more efficient. Our code is open source and documented at https://github.com/constantinpape/torch-em/blob/main/vimunet.md.
△ Less
Submitted 14 May, 2024; v1 submitted 11 April, 2024;
originally announced April 2024.
-
Probabilistic Domain Adaptation for Biomedical Image Segmentation
Authors:
Anwai Archit,
Constantin Pape
Abstract:
Segmentation is a key analysis tasks in biomedical imaging. Given the many different experimental settings in this field, the lack of generalization limits the use of deep learning in practice. Domain adaptation is a promising remedy: it trains a model for a given task on a source dataset with labels and adapts it to a target dataset without additional labels. We introduce a probabilistic domain a…
▽ More
Segmentation is a key analysis tasks in biomedical imaging. Given the many different experimental settings in this field, the lack of generalization limits the use of deep learning in practice. Domain adaptation is a promising remedy: it trains a model for a given task on a source dataset with labels and adapts it to a target dataset without additional labels. We introduce a probabilistic domain adaptation method, building on self-training approaches and the Probabilistic UNet. We use the latter to sample multiple segmentation hypothesis to implement better pseudo-label filtering. We further study joint and separate source-target training strategies and evaluate our method on three challenging domain adaptation tasks for biomedical segmentation.
△ Less
Submitted 21 March, 2023;
originally announced March 2023.
-
Biomedical image analysis competitions: The state of current participation practice
Authors:
Matthias Eisenmann,
Annika Reinke,
Vivienn Weru,
Minu Dietlinde Tizabi,
Fabian Isensee,
Tim J. Adler,
Patrick Godau,
Veronika Cheplygina,
Michal Kozubek,
Sharib Ali,
Anubha Gupta,
Jan Kybic,
Alison Noble,
Carlos Ortiz de Solórzano,
Samiksha Pachade,
Caroline Petitjean,
Daniel Sage,
Donglai Wei,
Elizabeth Wilden,
Deepak Alapatt,
Vincent Andrearczyk,
Ujjwal Baid,
Spyridon Bakas,
Niranjan Balu,
Sophia Bano
, et al. (331 additional authors not shown)
Abstract:
The number of international benchmarking competitions is steadily increasing in various fields of machine learning (ML) research and practice. So far, however, little is known about the common practice as well as bottlenecks faced by the community in tackling the research questions posed. To shed light on the status quo of algorithm development in the specific field of biomedical imaging analysis,…
▽ More
The number of international benchmarking competitions is steadily increasing in various fields of machine learning (ML) research and practice. So far, however, little is known about the common practice as well as bottlenecks faced by the community in tackling the research questions posed. To shed light on the status quo of algorithm development in the specific field of biomedical imaging analysis, we designed an international survey that was issued to all participants of challenges conducted in conjunction with the IEEE ISBI 2021 and MICCAI 2021 conferences (80 competitions in total). The survey covered participants' expertise and working environments, their chosen strategies, as well as algorithm characteristics. A median of 72% challenge participants took part in the survey. According to our results, knowledge exchange was the primary incentive (70%) for participation, while the reception of prize money played only a minor role (16%). While a median of 80 working hours was spent on method development, a large portion of participants stated that they did not have enough time for method development (32%). 25% perceived the infrastructure to be a bottleneck. Overall, 94% of all solutions were deep learning-based. Of these, 84% were based on standard architectures. 43% of the respondents reported that the data samples (e.g., images) were too large to be processed at once. This was most commonly addressed by patch-based training (69%), downsampling (37%), and solving 3D analysis tasks as a series of 2D tasks. K-fold cross-validation on the training set was performed by only 37% of the participants and only 50% of the participants performed ensembling based on multiple identical models (61%) or heterogeneous models (39%). 48% of the respondents applied postprocessing steps.
△ Less
Submitted 12 September, 2023; v1 submitted 16 December, 2022;
originally announced December 2022.
-
Stateless actor-critic for instance segmentation with high-level priors
Authors:
Paul Hilt,
Maedeh Zarvandi,
Edgar Kaziakhmedov,
Sourabh Bhide,
Maria Leptin,
Constantin Pape,
Anna Kreshuk
Abstract:
Instance segmentation is an important computer vision problem which remains challenging despite impressive recent advances due to deep learning-based methods. Given sufficient training data, fully supervised methods can yield excellent performance, but annotation of ground-truth data remains a major bottleneck, especially for biomedical applications where it has to be performed by domain experts.…
▽ More
Instance segmentation is an important computer vision problem which remains challenging despite impressive recent advances due to deep learning-based methods. Given sufficient training data, fully supervised methods can yield excellent performance, but annotation of ground-truth data remains a major bottleneck, especially for biomedical applications where it has to be performed by domain experts. The amount of labels required can be drastically reduced by using rules derived from prior knowledge to guide the segmentation. However, these rules are in general not differentiable and thus cannot be used with existing methods. Here, we relax this requirement by using stateless actor critic reinforcement learning, which enables non-differentiable rewards. We formulate the instance segmentation problem as graph partitioning and the actor critic predicts the edge weights driven by the rewards, which are based on the conformity of segmented instances to high-level priors on object shape, position or size. The experiments on toy and real datasets demonstrate that we can achieve excellent performance without any direct supervision based only on a rich set of priors.
△ Less
Submitted 21 March, 2023; v1 submitted 6 July, 2021;
originally announced July 2021.
-
Sparse Object-level Supervision for Instance Segmentation with Pixel Embeddings
Authors:
Adrian Wolny,
Qin Yu,
Constantin Pape,
Anna Kreshuk
Abstract:
Most state-of-the-art instance segmentation methods have to be trained on densely annotated images. While difficult in general, this requirement is especially daunting for biomedical images, where domain expertise is often required for annotation and no large public data collections are available for pre-training. We propose to address the dense annotation bottleneck by introducing a proposal-free…
▽ More
Most state-of-the-art instance segmentation methods have to be trained on densely annotated images. While difficult in general, this requirement is especially daunting for biomedical images, where domain expertise is often required for annotation and no large public data collections are available for pre-training. We propose to address the dense annotation bottleneck by introducing a proposal-free segmentation approach based on non-spatial embeddings, which exploits the structure of the learned embedding space to extract individual instances in a differentiable way. The segmentation loss can then be applied directly to instances and the overall pipeline can be trained in a fully- or weakly supervised manner. We consider the challenging case of positive-unlabeled supervision, where a novel self-supervised consistency loss is introduced for the unlabeled parts of the training data. We evaluate the proposed method on 2D and 3D segmentation problems in different microscopy modalities as well as on the Cityscapes and CVPPP instance segmentation benchmarks, achieving state-of-the-art results on the latter. The code is available at: https://github.com/kreshuklab/spoco
△ Less
Submitted 4 April, 2022; v1 submitted 26 March, 2021;
originally announced March 2021.
-
Proposal-Free Volumetric Instance Segmentation from Latent Single-Instance Masks
Authors:
Alberto Bailoni,
Constantin Pape,
Steffen Wolf,
Anna Kreshuk,
Fred A. Hamprecht
Abstract:
This work introduces a new proposal-free instance segmentation method that builds on single-instance segmentation masks predicted across the entire image in a sliding window style. In contrast to related approaches, our method concurrently predicts all masks, one for each pixel, and thus resolves any conflict jointly across the entire image. Specifically, predictions from overlap** masks are com…
▽ More
This work introduces a new proposal-free instance segmentation method that builds on single-instance segmentation masks predicted across the entire image in a sliding window style. In contrast to related approaches, our method concurrently predicts all masks, one for each pixel, and thus resolves any conflict jointly across the entire image. Specifically, predictions from overlap** masks are combined into edge weights of a signed graph that is subsequently partitioned to obtain all final instances concurrently. The result is a parameter-free method that is strongly robust to noise and prioritizes predictions with the highest consensus across overlap** masks. All masks are decoded from a low dimensional latent representation, which results in great memory savings strictly required for applications to large volumetric images. We test our method on the challenging CREMI 2016 neuron segmentation benchmark where it achieves competitive scores.
△ Less
Submitted 10 September, 2020;
originally announced September 2020.
-
The Semantic Mutex Watershed for Efficient Bottom-Up Semantic Instance Segmentation
Authors:
Steffen Wolf,
Yuyan Li,
Constantin Pape,
Alberto Bailoni,
Anna Kreshuk,
Fred A. Hamprecht
Abstract:
Semantic instance segmentation is the task of simultaneously partitioning an image into distinct segments while associating each pixel with a class label. In commonly used pipelines, segmentation and label assignment are solved separately since joint optimization is computationally expensive. We propose a greedy algorithm for joint graph partitioning and labeling derived from the efficient Mutex W…
▽ More
Semantic instance segmentation is the task of simultaneously partitioning an image into distinct segments while associating each pixel with a class label. In commonly used pipelines, segmentation and label assignment are solved separately since joint optimization is computationally expensive. We propose a greedy algorithm for joint graph partitioning and labeling derived from the efficient Mutex Watershed partitioning algorithm. It optimizes an objective function closely related to the Symmetric Multiway Cut objective and empirically shows efficient scaling behavior. Due to the algorithm's efficiency it can operate directly on pixels without prior over-segmentation of the image into superpixels. We evaluate the performance on the Cityscapes dataset (2D urban scenes) and on a 3D microscopy volume. In urban scenes, the proposed algorithm combined with current deep neural networks outperforms the strong baseline of `Panoptic Feature Pyramid Networks' by Kirillov et al. (2019). In the 3D electron microscopy images, we show explicitly that our joint formulation outperforms a separate optimization of the partitioning and labeling problems.
△ Less
Submitted 29 December, 2019;
originally announced December 2019.
-
Synthetic patches, real images: screening for centrosome aberrations in EM images of human cancer cells
Authors:
Artem Lukoyanov,
Isabella Haberbosch,
Constantin Pape,
Alwin Kraemer,
Yannick Schwab,
Anna Kreshuk
Abstract:
Recent advances in high-throughput electron microscopy imaging enable detailed study of centrosome aberrations in cancer cells. While the image acquisition in such pipelines is automated, manual detection of centrioles is still necessary to select cells for re-imaging at higher magnification. In this contribution we propose an algorithm which performs this step automatically and with high accuracy…
▽ More
Recent advances in high-throughput electron microscopy imaging enable detailed study of centrosome aberrations in cancer cells. While the image acquisition in such pipelines is automated, manual detection of centrioles is still necessary to select cells for re-imaging at higher magnification. In this contribution we propose an algorithm which performs this step automatically and with high accuracy. From the image labels produced by human experts and a 3D model of a centriole we construct an additional training set with patch-level labels. A two-level DenseNet is trained on the hybrid training data with synthetic patches and real images, achieving much better results on real patient data than training only at the image-level. The code can be found at https://github.com/kreshuklab/centriole_detection.
△ Less
Submitted 27 August, 2019;
originally announced August 2019.
-
GASP, a generalized framework for agglomerative clustering of signed graphs and its application to Instance Segmentation
Authors:
Alberto Bailoni,
Constantin Pape,
Nathan Hütsch,
Steffen Wolf,
Thorsten Beier,
Anna Kreshuk,
Fred A. Hamprecht
Abstract:
We propose a theoretical framework that generalizes simple and fast algorithms for hierarchical agglomerative clustering to weighted graphs with both attractive and repulsive interactions between the nodes. This framework defines GASP, a Generalized Algorithm for Signed graph Partitioning, and allows us to explore many combinations of different linkage criteria and cannot-link constraints. We prov…
▽ More
We propose a theoretical framework that generalizes simple and fast algorithms for hierarchical agglomerative clustering to weighted graphs with both attractive and repulsive interactions between the nodes. This framework defines GASP, a Generalized Algorithm for Signed graph Partitioning, and allows us to explore many combinations of different linkage criteria and cannot-link constraints. We prove the equivalence of existing clustering methods to some of those combinations and introduce new algorithms for combinations that have not been studied before. We study both theoretical and empirical properties of these combinations and prove that some of these define an ultrametric on the graph. We conduct a systematic comparison of various instantiations of GASP on a large variety of both synthetic and existing signed clustering problems, in terms of accuracy but also efficiency and robustness to noise. Lastly, we show that some of the algorithms included in our framework, when combined with the predictions from a CNN model, result in a simple bottom-up instance segmentation pipeline. Going all the way from pixels to final segments with a simple procedure, we achieve state-of-the-art accuracy on the CREMI 2016 EM segmentation benchmark without requiring domain-specific superpixels.
△ Less
Submitted 3 June, 2022; v1 submitted 27 June, 2019;
originally announced June 2019.
-
Leveraging Domain Knowledge to Improve Microscopy Image Segmentation with Lifted Multicuts
Authors:
Constantin Pape,
Alex Matskevych,
Adrian Wolny,
Julian Hennies,
Giula Mizzon,
Marion Louveaux,
Jacob Musser,
Alexis Maizel,
Detlev Arendt,
Anna Kreshuk
Abstract:
The throughput of electron microscopes has increased significantly in recent years, enabling detailed analysis of cell morphology and ultrastructure. Analysis of neural circuits at single-synapse resolution remains the flagship target of this technique, but applications to cell and developmental biology are also starting to emerge at scale. The amount of data acquired in such studies makes manual…
▽ More
The throughput of electron microscopes has increased significantly in recent years, enabling detailed analysis of cell morphology and ultrastructure. Analysis of neural circuits at single-synapse resolution remains the flagship target of this technique, but applications to cell and developmental biology are also starting to emerge at scale. The amount of data acquired in such studies makes manual instance segmentation, a fundamental step in many analysis pipelines, impossible. While automatic segmentation approaches have improved significantly thanks to the adoption of convolutional neural networks, their accuracy still lags behind human annotations and requires additional manual proof-reading. A major hindrance to further improvements is the limited field of view of the segmentation networks preventing them from exploiting the expected cell morphology or other prior biological knowledge which humans use to inform their segmentation decisions. In this contribution, we show how such domain-specific information can be leveraged by expressing it as long-range interactions in a graph partitioning problem known as the lifted multicut problem. Using this formulation, we demonstrate significant improvement in segmentation accuracy for three challenging EM segmentation problems from neuroscience and cell biology.
△ Less
Submitted 5 August, 2019; v1 submitted 25 May, 2019;
originally announced May 2019.
-
The Mutex Watershed and its Objective: Efficient, Parameter-Free Graph Partitioning
Authors:
Steffen Wolf,
Alberto Bailoni,
Constantin Pape,
Nasim Rahaman,
Anna Kreshuk,
Ullrich Köthe,
Fred A. Hamprecht
Abstract:
Image partitioning, or segmentation without semantics, is the task of decomposing an image into distinct segments, or equivalently to detect closed contours. Most prior work either requires seeds, one per segment; or a threshold; or formulates the task as multicut / correlation clustering, an NP-hard problem. Here, we propose an efficient algorithm for graph partitioning, the "Mutex Watershed''. U…
▽ More
Image partitioning, or segmentation without semantics, is the task of decomposing an image into distinct segments, or equivalently to detect closed contours. Most prior work either requires seeds, one per segment; or a threshold; or formulates the task as multicut / correlation clustering, an NP-hard problem. Here, we propose an efficient algorithm for graph partitioning, the "Mutex Watershed''. Unlike seeded watershed, the algorithm can accommodate not only attractive but also repulsive cues, allowing it to find a previously unspecified number of segments without the need for explicit seeds or a tunable threshold. We also prove that this simple algorithm solves to global optimality an objective function that is intimately related to the multicut / correlation clustering integer linear programming formulation. The algorithm is deterministic, very simple to implement, and has empirically linearithmic complexity. When presented with short-range attractive and long-range repulsive cues from a deep neural network, the Mutex Watershed gives the best results currently known for the competitive ISBI 2012 EM segmentation benchmark.
△ Less
Submitted 19 April, 2021; v1 submitted 25 April, 2019;
originally announced April 2019.
-
Synaptic Cleft Segmentation in Non-Isotropic Volume Electron Microscopy of the Complete Drosophila Brain
Authors:
Larissa Heinrich,
Jan Funke,
Constantin Pape,
Juan Nunez-Iglesias,
Stephan Saalfeld
Abstract:
Neural circuit reconstruction at single synapse resolution is increasingly recognized as crucially important to decipher the function of biological nervous systems. Volume electron microscopy in serial transmission or scanning mode has been demonstrated to provide the necessary resolution to segment or trace all neurites and to annotate all synaptic connections.
Automatic annotation of synaptic…
▽ More
Neural circuit reconstruction at single synapse resolution is increasingly recognized as crucially important to decipher the function of biological nervous systems. Volume electron microscopy in serial transmission or scanning mode has been demonstrated to provide the necessary resolution to segment or trace all neurites and to annotate all synaptic connections.
Automatic annotation of synaptic connections has been done successfully in near isotropic electron microscopy of vertebrate model organisms. Results on non-isotropic data in insect models, however, are not yet on par with human annotation.
We designed a new 3D-U-Net architecture to optimally represent isotropic fields of view in non-isotropic data. We used regression on a signed distance transform of manually annotated synaptic clefts of the CREMI challenge dataset to train this model and observed significant improvement over the state of the art.
We developed open source software for optimized parallel prediction on very large volumetric datasets and applied our model to predict synaptic clefts in a 50 tera-voxels dataset of the complete Drosophila brain. Our model generalizes well to areas far away from where training data was available.
△ Less
Submitted 7 May, 2018;
originally announced May 2018.
-
Resonance enhanced isotope-selective photoionization of YbI for ion trap loading
Authors:
M. Johanning,
A. Braun,
D. Eiteneuer,
Chr. Paape,
Chr. Balzer,
W. Neuhauser,
Chr. Wunderlich
Abstract:
Neutral Ytterbium (YbI) and singly ionized Ytterbium (YbII) is widely used in experiments in quantum optics, metrology and quantum information science. We report on the investigation of isotope selective two-photoionisation of YbI that allows for efficient loading of ion traps with YbII. Results are presented on two-colour (399 nm and 369 nm) and single-colour (399 nm) photoionisation and their ef…
▽ More
Neutral Ytterbium (YbI) and singly ionized Ytterbium (YbII) is widely used in experiments in quantum optics, metrology and quantum information science. We report on the investigation of isotope selective two-photoionisation of YbI that allows for efficient loading of ion traps with YbII. Results are presented on two-colour (399 nm and 369 nm) and single-colour (399 nm) photoionisation and their efficiency is compared to electron impact ionisation. Nearly deterministic loading of a desired number of YbII ions into a linear Paul trap is demonstrated.
△ Less
Submitted 17 June, 2010; v1 submitted 6 December, 2007;
originally announced December 2007.
-
Electrodynamically trapped Yb+ ions for quantum information processing
Authors:
Chr. Balzer,
A. Braun,
T. Hannemann,
Chr. Paape,
M. Ettler,
W. Neuhauser,
Chr. Wunderlich
Abstract:
Highly efficient, nearly deterministic, and isotope selective generation of Yb$^+$ ions by 1- and 2-color photoionization is demonstrated. State preparation and state selective detection of hyperfine states in \ybodd is investigated in order to optimize the purity of the prepared state and to time-optimize the detection process. Linear laser cooled Yb$^+$ ion crystals ions confined in a Paul tra…
▽ More
Highly efficient, nearly deterministic, and isotope selective generation of Yb$^+$ ions by 1- and 2-color photoionization is demonstrated. State preparation and state selective detection of hyperfine states in \ybodd is investigated in order to optimize the purity of the prepared state and to time-optimize the detection process. Linear laser cooled Yb$^+$ ion crystals ions confined in a Paul trap are demonstrated. Advantageous features of different previous ion trap experiments are combined while at the same time the number of possible error sources is reduced by using a comparatively simple experimental apparatus. This opens a new path towards quantum state manipulation of individual trapped ions, and in particular, to scalable quantum computing.
△ Less
Submitted 3 February, 2006;
originally announced February 2006.