H UMAN
P ERCEPTION - BASED
E VALUATION
C RITERION
FOR
U LTRA - HIGH
R ESOLUTION
C ELL
M EMBRANE S EGMENTATION
Anonymous authors
Paper under double-blind review
A BSTRACT
Computer vision technology is widely used in biological and medical data analysis and understanding.
However, there are still two major bottlenecks in the ﬁeld of cell membrane segmentation, which seriously hinder further research: lack of sufﬁcient high-quality data and lack of suitable evaluation criteria.
In order to solve these two problems, this paper ﬁrst introduces an Ultra-high Resolution Image Segmentation dataset for the Cell membrane, called U-RISC, the largest annotated
Electron Microscopy (EM) dataset for the Cell membrane with multiple iterative annotations and uncompressed high-resolution raw data.
During the analysis process of the U-RISC, we found that the current popular segmentation evaluation criteria are inconsistent with human perception.
This interesting phenomenon is conﬁrmed by a subjective experiment involving twenty people.
Furthermore, to resolve this inconsistency, we propose a new evaluation criterion called Perceptual
Hausdorff Distance (PHD) to measure the quality of cell membrane segmentation results.
Detailed performance comparison and discussion of classic segmentation methods along with two iterative manual annotation results under existing evaluation criteria and PHD is given.
I NTRODUCTION
Electron Microscopy (EM) is a powerful tool to explore ultra-ﬁne structures in biological tissues, which has been widely used in the research areas of medicine and biology ( ERLANDSON (2009);
Curry et al.
; Harris et al.
In recent years, EM techniques have pioneered an emerging ﬁeld called “Connectomics” (Lichtman et al.
(2014)), which aims to scan and reconstruct the whole brain circuitry at the nanoscale.
“Connectomics” has played a key role in several ambitious projects, including the BRAIN Initiative ( Insel et al.
and MICrONS ( Gleeson & Sawyer (2018)) in the U.S., Brain/MINDS in Japan ( Dando (2020)), and the China Brain Project ( Poo et al.
Because EM scans brain slices at the nanoscale, it produces massive images with ultra-high resolution and inevitably leads to the explosion of data.
However, compared to the advances of EM, techniques of data analysis fall far behind.
In particular, how to automatically extract information from massive raw data to reconstruct the circuitry map has growingly become the bottleneck of EM applications.
One critical step in automatic EM data analysis is Membrane segmentation.
With the introduction of deep learning techniques, signiﬁcant improvements have been achieved in several public available
EM datasets ISBI 2012 and SNEMI3D ( ISBI 2012 (2012); ISBI 2013 (2013); Arganda-Carreras et al. (2015b)
; Lee et al. One of the earliest works ( Ciresan et al.
(2012) used a succession of max-pooling convolutional networks as a pixel classiﬁer, which estimated the probability of a pixel is a membrane.
Ronneberger et al.
(2015) presented a U-net structure with contracting paths, which captures multi-contextual information.
Fully convolutional networks (FCNs) proposed by Long et al. (2015) led to a breakthrough in semantic segmentation.
Follow-up works based on Unet and FCN structure ( Xie & Tu (2015); Drozdzal et al.
; Hu et al.
; Zhou et al.
(2018); Chaurasia & Culurciello (2017); Yu et al.
; Chen et al.
(2019b)) have also achieved outstanding results near-human performance.
Despite much progress that has been made in cell membrane segmentation for EM data thanks to deep learning, one risk to these popular and classic methods is the dataset such as ISBI 2012 and SNEMI3D, which employed compressed images instead of the original ultra-high resolution images.
Considering that image compression generally loses many texture details, how can these classic deep learning based segmentation methods work on EM images with the original resolution?
Moreover, how robust of these methods when they are compared with human performance on original EM images?
To expand the research of membrane segmentation on more comprehensive EM data, we ﬁrst established a dataset “U-RISC” containing images with original resolution (10000 × 10000, Fig. 1).
To ensure the quality of annotation, it also costs us over 10,000 labor hours to label and doublecheck the data.
To the best of our knowledge, U-RISC is the largest uncompressed annotated and EM dataset today.
Next, we tested several classic deep learning based segmentation methods on U-RISC and compared the results to human performance.
We found that the performance of these methods was much lower than that of the ﬁrst annotation.
To understand why human perception is better than the popular segmentation methods, we examined in detail the Membrane segmentation results by these popular segmentation methods. How to measure the similarity between two image segmentation results has been widely discussed ( Yeghiazaryan & Voiculescu (2018); Niessen et al. (2000); Veltkamp & Hagedoorn (2000); Lee et al.
Varduhi Yeghiazaryan ( Yeghiazaryan & Voiculescu (2018)) discussed the family of boundary overlap metrics for the evaluation of medical image segmentation. Veltkamp. etc ( Veltkamp & Hagedoorn (2000)) formulated and summed up the similarity measures in a more general condition. In some challenges, such as ISBI ( Arganda-Carreras et al.
(2015a))
, they also considered multiple metrics like Rand score on both original images and thinned images.
However, we found there was a certain inconsistency between current most popular evaluation criteria for segmentation(e.g.
F1 score, IoU) and human perception: while some ﬁgures were rated signiﬁcantly lower in F1 score or IoU, they were “perceived” better by humans (Fig. 4).
Such inconsistency motivated us to propose a human-perception based criterion, Perceptual Hausdorff Distance (PHD) to evaluate image qualities.
Further, we set up a subjective experiment to collect human perception about membrane segmentation, and we found the PHD criteria is more consistent with human choices than traditional evaluation criteria.
Finally, we found the current popular and classical segmentation methods need to be revisited with PHD criteria.
Overall, our contribution in this work lies mainly in the following two parts: (1) we established the largest, original image resolution-based EM dataset for training and testing; (2) we proposed a human-perception based evaluation criterion, PHD, and veriﬁed the superiority of PHD by subjective experiments.
The dataset we contributed and the PHD criterion we proposed may help researchers to gain insights into the difference between human perception and conventional evaluation criteria, thus motivate the further design of the segmentation method to catch up with the human performance on original EM images.
U-RISC: U LTRA - HIGH R ESOLUTION I MAGE S EGMENTATION DATASET FOR C ELL MEMBRANE
Supervised learning methods rely heavily on high-quality datasets.
To alleviate the lack of training data for cell membrane segmentation, we proposed an Ultra-high Resolution Image Segmentation dataset for Cell membrane, called U-RISC.
The dataset was annotated upon RC1, a large scale retinal serial section transmission electron microscopic (ssTEM) dataset, publically available upon request and described in the work of Anderson et al.
The original RC1 dataset is a 0.25mm diameter, 370 TEM slices volume, spanning the inner nuclear, inner plexiform, and ganglion cell layers, acquired at 2.18 nm/pixel across both axes and 70nm thickness in z-axis. From the 370 serialsection volume, we clipped out 120 images in the size of 10000*10000 pixel from randomly chosen sections.
Then, we manually annotated the cell membranes in an iterative annotation-correction procedure.
Since the human labeling process is very valuable for uncovering the human leaning process, during the relabeling process, we reserved the intermediate results for public release.
The U-RISC dataset will be released on https://Anonymous.com on acceptance.
2. C OMPARISON W ITH O THER D ATASETS
ISBI 2012 (Cardona et al.
(2010)) published a set of 30 images for training, which were captured from the ventral nerve cord of a Drosophila ﬁrst instar larva at a resolution of 4 × 4 × [Figure] [Figure] nm/pixel through ssTEM (Arganda-Carreras et al.
(2015b) ; ISBI 2012 (2012)).
Each image contains 512 × 512 pixels, spanning a realistic area of 2 × 2 µ m approximately. In the challenge of SNEMI3D (Kasthuri et al. ; ISBI 2013 (2013)), the training data is a 3D stack of 100 images in the size of 1024 × 1024 with the voxel resolution of 6 × 6 × 29 nm/pixel. The raw images were acquired at the resolution of 3 × 3 × 29 nm/pixel using serial section scanning electron microscopy (ssSEM) from mouse somatosensory cortex (Kasthuri et al.
; ISBI 2013 (2013)). URISC containspieces of 10000 × 10000 pixels annotated images at the resolution of 2.18 × 2.18 × 70 nm/pixel from rabbit retina.
Due to the difference of species and tissue, URISC can ﬁll in the blank of annotated vertebrate retinal segmentation dataset. Besides that, URISC has some other characteristics which can be focused on in the future segmentation study. The ﬁrst one is that the image size and realistic size of URISC is much larger, 400 (100) and 100 (9) times compared with ISBI (SNEMI3D) respectively (Fig.(c)), which can be applied in developing deep learning based segmentation methods according to various demands.
And along with the iterative annotation procedure URISC actually contains 3 sets of annotation result with increasing accuracy, which could serve as ground truth at different level standard.
And the total number of the pair of image and annotation is 12 and 3.6 times than ISBI and SNEMI3D respectively (Fig.
1 (d)).
An example of the image with its label is shown in the Supplementary.
Due to the limitation of the size of the supplementary material, we only uploaded a quarter (5000 pixel × 5000 pixel) size of the original image with its label.
[Figure]
2. T RIPLE L ABELING P ROCESS
The character of high resolution in TEM image can display a much more detailed sub-cellular structure, which requests more patience to label out the cell (Fig. 2(a)).
Besides, the imaging quality can be affected by many factors, such as section thickness or sample staining (Fig. 2(b)).
And low imaging quality also requests more labeling efforts.
Therefore, increasing labeling efforts is essential to completely annotate U-RISC.
To guarantee the labeling accuracy, we set up an iterative correction mechanism in the labeling process (Fig. 3).
Before starting the annotation, labeling rules were introduced to all annotators.
58 qualiﬁed annotators were allowed to participate in the ﬁnal labeling process.
After the ﬁrst round annotation, 5 experienced lab staff with sufﬁcient background knowledge were responsible to point out labeling errors pixel by pixel during the second and the third rounds of annotation.
Finally, the third round annotation results were regarded as the ﬁnal “ground truth”.
And previous two rounds of manual annotations are also saved for later analysis.
Fig.shows an example of the two inspection processes.
We can see that there are quiet a few mislabeled and missed labeled cell membranes in each round.
Therefore, the iterative correction mechanism is very necessary.
P ERCEPTION - BASED E VALUATION
In the analysis of EM data, membrane segmentation is generally an indispensable key step.
However, in the ﬁeld of cell membrane segmentation, most of the previous studies, such as Zhou et al.
; Chaurasia & Culurciello (2017); Drozdzal et al.
(2016), were not speciﬁcally designed for high resolution datasets such as U-RISC.
In addition, although many researchers discussed various evaluation criteria for medical and general tasks, few researchers actually incorporate them into the design of the architectures of cell membrane segmentation methods.
By comparing the segmentation results of the popular and classic segmentation methods, we found that the widely used evaluation criteria of segmentation were inconsistent with human perception in some cases, which is further discussed through the perceptual consistency experiment (details in Sec. 3.2).
To address this issue, we proposed a new evaluation criterion called Perceptual Hausdorff Distance (PHD).
The experimental results showed that it was more consistent with human perception.
3. I NCONSISTENCY BETWEEN E XISTING E VALUATION C RITERIA AND P ERCEPTION
Many researchers have proposed various metrics for segmentation evaluation ( Yeghiazaryan & Voiculescu (2018); Niessen et al. (2000); Veltkamp & Hagedoorn (2000); Arganda-Carreras et al. (2015b)
; Lee et al.
Some of them, which are the most popular, such as F1 score / Dice
Coefﬁcient and IoU ( Sasaki et al.
Dice (1945); Kosub (2019)) are used as the evaluations in most segmentation methods ( Ronneberger et al.
; Zhou et al.
(2018); Chaurasia & Culurciello (2017);
Yu et al.
; Chen et al.
(2019b)). In ISBI2012 cell segmentation challenge used Rand scores (V-Rand and V-Info) ( Arganda-Carreras et al. (2015b)) on thinned membrane for evaluation.
Recently, researchers made discussions on various boundary overlap metrics for the evaluation of medical image segmentation (Yeghiazaryan & Voiculescu (2018)).
The most popular evaluation criteria, such as F1 score, are based on the statistics of the degree to which pixels are classiﬁed correctly. However, quality of segmentation should be judged with respect to the ultimate goal.
When we need to use segmentation to reconstruct the whole structure of membranes and connect them, such statistics may not be consistent with human perception in cell membrane segmentation tasks.
There are also some metrics designed based on point set distance, such as ASSD (Yeghiazaryan & Voiculescu (2018)), which is not widely used in recent deep learning researches.
In the process of segmentation experiments, some interesting phenomena were found.
Fig.
4 shows an example of the original image with its manual annotation and segmentation results by two methods GLNet ( Chen et al.
(2019b)) and U-Net ( Ronneberger et al.
The scores indicated that (d) was more similar to (b) than (c).
It should be noted that when we looked at the area surrounded by the red dashed lines in the images, It should be noted that if these segmentation result are used for reconstructing the structure of cells, the mistakes and loss of structure will be more noticeable when subjects inspect the area surrounded by the red dashed lines in the images, Therefore we consider that (c) is a better prediction, because (d) misses some edges.
The reason for the three scores of (c) are lower was that the predicted cell membrane of (c) was thicker than manual labeling.
Therefore, it can be inferred that the existing evaluation criteria might not sufﬁciently robust to variations in the thickness and structures of the membrane, and the evaluation result was more likely inconsistent with human perception.
[Figure]
3. P ERCEPTUAL C ONSISTENCY E XPERIMENTS
In order to verify the above conjecture, a subjective experiment was designed to explore the consistency with the existing evaluation criteria and human subjective perception.
Six popular and classical segmentation methods were used to generate cell membrane segmentation results on URISC: U-net ( Ronneberger et al.
, LinkNet( Chaurasia & Culurciello (2017)), CASENet ( Yu et al.
SENet ( Hu et al.
U-Net++ ( Zhou et al.
(2018)),and GLNet ( Chen et al. (2019b)).
Using these segmentation results, 200 groups of images were randomly selected.
Each group contained 3 images: the ﬁnal manual annotation (ground truth) and two automatically generated segmentation results for the same input cell image.
20 subjects were recruited to participate in the experiments.
They had either a biological background or experience in cell membrane segmentation and reconstruction.
For each group, each of thesubjects had three choices.
If the subject can tell which segmentation result is more similar to the ground truth, he or she can choose which one.
Otherwise, the subject can choose “Difﬁcult to choose”.
The experiment interface is shown in the Appendix I.
Before the experiment, the subjects were trained on the purpose and source of the images.
During the experiment, 200 groups of images were divided into four groups on average in order to prevent the subjects from choosing randomly due to fatigue.
For each batch of groups, the subjects needed to complete the judgment continuously without interruption.
After the experiment, for each group, if there were more than 10 votes of the same number, it was called a valid group.
Otherwise, it was invalid and discarded.
There were a total of 113 valid groups.
Then, based on these valid groups, the consistency of the F1 score, IoU, and Dice with human choices was calculated.
According to our experimental results, the consistency of F1 score, IoU, and Dice with human choice was only 34.51%, 35.40%, and 34.51%, respectively.
Therefore, it can be inferred that the three criteria are not consistent with human subjective perception in most cases.
More results and design of subjective experiments are shown in the Appendix II, and IV,.
3. P ERCEPTUAL H AUSDORFF D ISTANCE
Based on the subjective experimental results, it was veriﬁed that the widely used evaluation criteria for general segmentation were inconsistent with human perception of cell membrane segmentation.
This paper proposes a new evaluation standard based on human perception, namely, Perceptual
Hausdorff Distance (PHD for short) , considering the structure but ignoring the thickness of cell membrane.
An Overview of PHD.
As Fig.
4 shows, from the perspective of neuronal reconstruction, the thickness of the cell membrane is not the key for evaluation.
In fact, when the goal is to reconstruct the structure of cells, humans will pay more attention on structure changes, instead of thickness changes.
Hence, when measuring the similarity of two cell membrane segmentation results, in order to eliminate the inﬂuence of thickness, the segmentation results of two cell membranes were skeletonized, and then the distance between two skeletons was calculated to measure the difference.
Since the skeleton is a collection of different points, and Hausdorff distance is a common distance to calculate the difference between two sets of points, the proposed PHD is built upon Hausdorff distance.
On the other hand, through subjective experiments, it was found that people tend to ignore the slight offset between the membrane.
Therefore, based on the above two considerations, the Perceptual
Hausdorff Distance (PHD) based on Hausdorff distance( Huttenlocher et al. (1993);
Aspert et al.
(2002); Rachasingho & Tasena (2020)) with modiﬁcation was designed.
Fig.
5 shows the overview of PHD.
The details are as follows. [Figure]
Skeleton
[Figure]
Segmentation
Skeleton
[Figure] Then, the two inputs are skeletonized. Finally, a distance can be calculated between two skeletons with different tolerance distances are two toy cases for intuitively understanding the inﬂuence of tolerance distance in Step 1.
Skeletonize the cell membrane.
Zhang-Suen thinning algorithm is used ( Zhang & Suen (1984)
; Saudagar & Mohammed (2016)) and re-implemented to obtain the skeleton of the cell membrane.
Step 2.
Calculate the distance between skeletons.
Hausdorff distance is a common distance used to calculate the difference between two point sets.
Consider two unordered nonempty sets of points X and Y and the Euclidean distance d ( x , y ) between two point sets.
The Hausdorff distance between X and Y is deﬁned as
[Equation]
which can be understood as the maximum value of the shortest distance from a point set to another point set.
It is easy to prove that the Hausdorff distance is a metric ( Choi (2019)).
In the task of cell membrane segmentation, we should pay attention to the global distance between two point sets, while Hausdorff distance is sensitive to outliers in two point sets.
Therefore, the average distance of the two point sets is obtained naturally by using the average operation instead of all the max operations.
Furthermore, it was found that people have tolerance for the small offset between segmentation results.
Speciﬁcally, if the distance between two points is very small, people tend to ignore it.
Therefore, a concept called Tolerance Distance t is deﬁned, which represents human tolerance for small errors.
The Perceptual Hausdorff Distance (PHD) is deﬁned as Eq. d
[Equation]
To intuitively understand the inﬂuence of tolerance distance in PHD, toy cases (a) and (b) as shown in Fig.
5 are taken as examples.
In case (a), the blue skeleton scored 19 points while the orange one scored 18 points.
Two skeletons are close in the Euclidean Space but do not coincide.
Among all the
Euclidean distance d ( x , y ) of x ∈ X and y ∈ Y , the max distance is 2 pixels, and the most common distance is 1.
When t = 0 , which means no mistake can be tolerated, and the PHD is high.
If t = 1 , the PHD value drops a lot.
When the t = 2 , PHD becomes 0.
In case (b), there is a large offset between two skeletons.
When the t is set to [2 , 4] , the decline of PHD value is slow.
When t = 6 , it drops to 0, which is the max distance between two point sets of skeletons.
The Different settings of t represent the degree of tolerance to the distance between the two skeletons.
In practical applications, different tolerance distances can be adopted according to different situations.
Consistency between PHD and human perception.
The consistency with human perception of PHD and existing related criteria (TPVF, TNVF, Prec, RVD, Hausdorff, ASSD, V-Rand, and V-Info) based on the subjective experimental results were also calculated (as described in Section 3.2, the formulas can be found in Appendix V).
The result showed that compared with other criteria, PHD with appropriate tolerance distance was more consistent with human perception.
As shown in Fig.6, while tolerance distance t of PHD increasing from 0 to 800, its consistency to human perception rose ﬁrst and then dropped slowly to 0, suggesting human vision does have tolerance for certain offset.
Speciﬁcally, the maximum value can be reached at 65.48%, when tolerance distance t was set to 3, suggesting that our perceptions prefer to tolerant small perturbations.
It was worthy to note that the optimal PHD score (65.48%) was nearly double of the consistency scores obtained by pixel-error based metrics, such as F1 score.
Our experiment shows that most of these compared criteria in color bars can be improved by skeletonizing the segmentation before evaluation to a certain extent.
In Fig. 6, the consistency of these criteria with human perception calculated based on original images can only reach about 30%, while they can improve about 10% on skeletons.
Even the best of these metrics (ASSD) on skeletons can only achieve 52.43%, which is signiﬁcantly lower than the score of PHD performance with t = 10 . Therefore, it can be concluded that the PHD performs more consistently with human.
R E - EXAMINING PHD ON C LASSIC D EEP L EARNING BASED S EGMENTATION M ETHODS WITH U-RISC
In the previous two sections, we proposed a new ultra-high resolution cell membrane segmentation dataset U-RISC and a new perceptual criteria PHD to help solve the two bottlenecks in the ﬁeld of cell membrane segmentation.
The subjective experiment on a small-scale dataset demonstrated that
PHD is more consistent with human perception for the evaluation of cell membrane segmentation than some widely used criteria.
In order to understand the performance of deep learning methods on the U-RISC dataset, we conducted an in depth investigation on U-RISC with representative deep learning based segmentation methods and different evaluation criteria.
To be speciﬁc, we chose 6 representative algorithms (
U-net (Ronneberger et al.
, LinkNet(Chaurasia & Culurciello (2017)), CASENet (Yu et al.
(2017)), SENet (Hu et al.
U-Net++ (Zhou et al.
(2018)),and GLNet (Chen et al. (2019b)))
and re-implemented them on U-RISC dataset.
Then four evaluation criteria were used to compare the segmentation results: F1 score, IoU, TPVF, TNVF, Prec, RVD, Hausdorff, ASSD, V-Rand, and V-Info, and PHD.
As mentioned in Sec. 2, the results of the ﬁrst two rounds of manual labeling results are retained.
Therefore, the results of manual annotation under different evaluation criteria will also be analyzed.
Experiment Settings.
All the six methods use same training data and testing data to compare the performance.
And the parameters and loss functions are same as they are proposed in their references.
The parameters for each method and other details are shown in Appendix V.
Experiments Results.
The experiment results were shown in Table.
The table showed the scores of different evaluation criteria on the ﬁrst two rounds of manual annotation results and six segmentation results with the ground truth.
Our ﬁrst ﬁnding was that U-RISC was a challenging dataset in the ﬁeld of cell membrane segmentation.
As shown in Table.
1,the performance of deep learning based methods gained around 0.
in F1-scores, far below the human level (0.98-0.99) (the ﬁrst annotation performace) on U-RISC dataset, by contrast, they all exceeded 0.95 on the ISBI 2012.
Despite possible improvements by parameter tuning, to such ultra-high resolution images, there was clearly a huge gap between the current popular segmentation methods and human performance.
Our second ﬁnding was that evaluation rankings for F1-score, IoU, V-Rand-sk, and V-Info-sk were more consistent with each other, but different from PHD-based rankings. Speciﬁcally, while PHD tends to chooses CASNet, none of the other metrics chose CASNet as the best choice.
According to the subjective experimental results of Sec. 3.2, PHD was much closer to human perception.
Therefore, the change of ranking led by PHD may also inspire researchers to re-consider the evaluation criteria for cell membrane segmentation algorithms.
It also provides a new perspective for promoting the development of segmentation algorithms.
Discussion.
Based on the results of these six algorithms, it can be seen that LinkNet and CASENet are better than other methods.
From the perspective of network design, LinkNet makes full use of the low-level local information and directly connects the low-level encoder to the decoder of corresponding size.
This design pays more attention to the capture of local information, which leads to a more accurate local prediction.
CASENet takes full account of the continuity of the edge and makes the low-level features strengthen the high-level semantic information by jumping links between the low-level feature and high-level feature, which pays more attention to structural [Table]
information. Therefore, the design of LinkNet might be preferred by the traditional evaluation criteria, while CASENet might be preferred by the PHD.
This also explains why the two methods rank differently under these two types of evaluation criteria.
More local segmentation results of different algorithms are shown in the Appendix III.
In addition, as an example, we add a new experiment using U-Net in ISBI2012 and SNMI3D datasets (VI).
The results show that U-Net with our chosen parameters can perform close to SOTA on ISBI2012 (ours: V-Rand=0.9689,V-Info=0.9723; SOTA: V-Rand=0.9837, V-Info=0.9878 (on skeleton)) and SNEMI3D (ours:V-Rand=0.9389; SOTA: V-Rand=0.9751 (on skeleton)), although we made little efforts in parameter tuning.
However, with the same parameter setting, U-Net gets poor scores (V-Rand=0.5288, V-Info=0.5178) on U-RISC.
Such a big gap in its performance between U-RISC and previous datasets suggests the challenge from U-RISC dataset, which hopefully will motivate novel designs of machine learning methods in the future.
D ISCUSSION AND C ONCLUSION
This paper aims to solve the two bottlenecks in the development of cell membrane segmentation.
Firstly, we proposed U-RISC,Ultra-high Resolution Image Segmentation dataset for Cell membrane, the largest annotated EM dataset for the Cell membrane so far.
To our best knowledge, U-RISC is the only uncompressed annotated EM dataset with multiple iterative annotations and uncompressed high-resolution raw image data.
During the analysis process of the U-RISC, we found a certain inconsistency between current evaluation criteria for segmentation (e.g. F1 score, IoU) and human perception.
Therefore, this article secondly proposed a human-perception based evaluation criterion, called Perceptual Hausdorff Distance (PHD).
Through a subjective experiment on a smallscale dataset, experiments results demonstrated that the new criterion is more consistent with human perception for the evaluation of cell membrane segmentation.
In addition, the evaluation criteria of
PHD and existing classic deep learning segmentation methods are re-examined.
In future research, we will consider how to improve deep learning segmentation methods from the perspective of cell membrane structure and apply PHD criterion for connectomics research. More disccusions are shown in Appendix VII.
A A PPENDIX
I. Fig.
7 is the interface of perceptual consistency experiments
II.
Fig.
8 and Fig.
9 are some examples of subjective experiment images.
III.
Fig.
10 and Fig.
11 are some examples of segmentation results of different algorithms. 4.1 Subjective Experiment
1) Firstly, the 20 human raters were introduced the value of cell membrane segmentation to connectivity and the importance of structure before testing.
And then we used several simple examples to teach them about the experimental process.
During the formal experiment, the distribution and selection of data were random.
The subjects only need to choose one of the two images that they think is more similar to the ground truth. The 200 groups of images for subjective experiment were randomly selected from the segmentation results produced by the above six methods. Therefore, the training data were from the same dataset as those that were used to create the 200 groups of images that the 20 humans evaluated the results. In order to ensure the continuity of the experiment, each subject was asked to judge each group of images within a speciﬁed time (less than 10 minutes).
In order to prevent the subjects from fatigue in a long period of experiments, 200 groups of images were distributed to the subjects four times on average.
4.2 Experiments on U-RISC dataset.
All the six methods use same training data and testing data to compare the performance.
And the parameters and loss functions are same as they are proposed in their references.
In the training stage, 60% of the dataset was used as the training data, and then the original image was randomly cut into 1024 × 1024 patches to generate 50,000 training images and 20,000 validation images.
Random ﬂipping and clipping were used for data augmentation.
Four V100 GPUs were used to train each algorithm.
In the testing stage, the original image was cut into the same size of training image, and the patch was tested.
These patches were eventually spliced back to the original size for evaluation.
[Table]
V. Formulas of criteria mentioned in the texture.
The formulas of metrics we compared are shown in Table 3.
The symbols in formulations are explained as follows.
- Precision (also called positive predictive value) is the fraction of relevant instances among the retrieved instances, while recall (also known as sensitivity) is the fraction of the total amount of relevant instances that were actually retrieved.
- TP (true positives), TN (true negatives), FP (false positives), and FN(false negatives) compare the results of the classiﬁer under prediction with ground truth
The terms positive and negative refer to the classiﬁer’s prediction, and the terms true and false refer to whether that prediction corresponds to the ground truth.
- X and Y are two point sets.
d ( x , y ) are the points in X and Y respectively.
In V-Rand (Arganda-Carreras et al.
(2015b))
, suppose that S is the predicted segmentation and T is the ground truth segmentation.
Deﬁne p
i,j as the probability that a randomly chosen pixel belongs to segment i in S and segment j in T .
This joint probability distribution satisﬁes the normalization condition (cid:80) ij p ij = 1 .
The marginal distribution s i
= (cid:80) j p ij is the probability that a randomly chosen pixel belongs to segment i in S , and the marginal distribution t j = (cid:80) i p ij is deﬁned similarly.
In V-Info (Arganda-Carreras et al.
(2015b)), the mutual information
I ( S ; T ) = (cid:80) ij p ij log p ij − (cid:80)
i s i log
s
i − (cid:80) j t j log t j is a measure of similarity between S and T .
H ( S ) = − (cid:80)
i s i log
s i is the entropy function.
VI.
Experiments on ISBI2012 and SNEMI3D with U-Net.
(Table
VII.
Further Discussion.
Although the content of this paper is mainly involved in EM cell segmentation we think the signiﬁcance is beyond.
[Table]
According to experimental results
[Table]
the popular methods do not perform well on our dataset (exceeded 95% on ISBI
[Table]
while about 60% on U-RISC).
It shows that U-RISC is a challenging dataset that can promote the development of related machine learning and deep learning methods
[Table]
The U-RISC dataset may reveal several classic challenges in the ﬁeld that haven’t been solved:
One challenge might be the “imbalance problem of samples” ( Alejo et al.
; Li et al.
Zhang et al.
Due to ultra-high resolution images, the pixels of labeled cell membranes only account for 5.64% of total pixels in training sets, in contrast to 21.96% in ISBS2012 and 33.23% in
SNEMI3D. The future design of deep learning methods on U-RISC will have to solve this issue.
Some other challenges might include, e.g. ultra high-resolution image segmentation ( Demir et al.
; Zhao et al.
; Chen et al.
(2019a), appropriate loss function design ( Sudre et al.
Spiring (1993); Choromanska et al.
(2015)),and the issues related to ”unclosed” edges as suggested by the reviewer 3.
Taken together, we strongly believe that the U-RISC dataset will have great contribution in technique novelty, by revealing defects in the existing popular methods and promoting novel algorithms for solving classic challenges in machine learning or deep learning community.
In addition, the design of evaluation criteria has been widely concerned in the ﬁeld of computer science ( Gerl et al.
Lin et al. (2015); Liu et al.
The PHD we proposed may inspire researchers from a new perspective and further promote the developments of algorithms.
The technical novelties of the PHD metric lie in many aspects.
To list a few, (1) It can be potentially used
[Table]
in other tasks, such as vascular segmentation ( Gerl et al.
bone segmentation ( Lin et al.
, edge detection ( Liu et al.
(2018)), and other tasks related to structural and shape information.
For example, ( Gerl et al.
(2020)) successfully used a distance-based criterion to improve skin layer segmentation in optoacoustic images.
(2) It can be modiﬁed into loss functions which is also part of our on-going work.
It is worthy to note that some works have successfully integrated Hausdorff distance into the loss function ( Genovese et al.
(2012); Karimi & Salcudean (2019)
; Ribera et al.
[Figure] [Figure]
CASENet
GLNet
LinkNet
SENet
U-Net
U-Net++
Manual
Annotation
[Figure] [Figure]
CASENet
GLNet
LinkNet
SENet
U-Net
U-Net++
Manual
Annotation