Note
Click here to download the full example code
Weak Segments Performance#
This notebook provides an overview for using and understanding the weak segment performance check.
Structure:
What is the purpose of the check?#
The check is designed to easily identify the model’s weakest segments. The segments are characterized by the image properties such as contrast and aspect ratio.
Automatically detecting weak segments#
The check performs several steps:
We calculate the image properties for each sample. The properties to calculate can be passed explicitly or resort to the default image properties.
We calculate loss for each sample in the dataset using the provided model or predictions, the loss function can be passed explicitly or set to a default based on the task type.
We train multiple simple tree based models, each one is trained using two properties to predict the per sample error calculated before.
We convert each of the leafs in each of the trees into a segment and calculate the segment’s performance. For the weakest segments detected we also calculate the model’s performance on the data segments surrounding them.
Run the check#
from deepchecks.vision.checks import WeakSegmentsPerformance
from deepchecks.vision.datasets.detection import coco
coco_data = coco.load_dataset(train=False, object_type='VisionData')
model = coco.load_model()
check = WeakSegmentsPerformance()
result = check.run(coco_data, model)
result
Validating Input:
| | 0/1 [Time: 00:00]
Validating Input:
|#####| 1/1 [Time: 00:07]
Validating Input:
|#####| 1/1 [Time: 00:07]
Ingesting Batches:
| | 0/2 [Time: 00:00]
Ingesting Batches:
|##5 | 1/2 [Time: 00:08]
Ingesting Batches:
|#####| 2/2 [Time: 00:18]
Ingesting Batches:
|#####| 2/2 [Time: 00:18]
Computing Check:
| | 0/1 [Time: 00:00]/home/runner/work/deepchecks/deepchecks/venv/lib/python3.9/site-packages/joblib/externals/loky/process_executor.py:700: UserWarning:
A worker stopped while some jobs were given to the executor. This can be caused by a too short worker timeout or by a memory leak.
Computing Check:
|#####| 1/1 [Time: 00:10]
Computing Check:
|#####| 1/1 [Time: 00:10]
If you have a GPU, you can speed up this check by passing it as an argument to .run() as device=<your GPU>
To display the results in an IDE like PyCharm, you can use the following code:
# result.show_in_window()
The result will be displayed in a new window. %%
We see in the results that the check indeed found several segments on which the model performance is below average. In the heatmap display we can see the model’s performance on the weakest segments and their environment with respect to the two segmentation features. In order to get the full list of weak segments found we can look at the result.value attribute.
result.value['weak_segments_list']
Now we will run a check with properties and minimum segment size ratio (the minimal fraction of the data to be considered as a segment) different from the defaults.
from deepchecks.vision.utils.image_properties import brightness, texture_level
properties = [{'name': 'brightness', 'method': brightness, 'output_type': 'numerical'},
{'name': ' texture', 'method': texture_level, 'output_type': 'numerical'}]
check = WeakSegmentsPerformance(segment_minimum_size_ratio=0.03, image_properties=properties)
result = check.run(coco_data, model)
result.show_in_window()
Validating Input:
| | 0/1 [Time: 00:00]
Validating Input:
|#####| 1/1 [Time: 00:08]
Validating Input:
|#####| 1/1 [Time: 00:08]
Ingesting Batches:
| | 0/2 [Time: 00:00]
Ingesting Batches:
|##5 | 1/2 [Time: 00:08]
Ingesting Batches:
|#####| 2/2 [Time: 00:16]
Ingesting Batches:
|#####| 2/2 [Time: 00:16]
Computing Check:
| | 0/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:00]
Define a condition#
We can define on our check a condition that will validate that the ratio of the model performance on the weakest segment to the average model performance is less than a specified ratio.
Validating Input:
| | 0/1 [Time: 00:00]
Validating Input:
|#####| 1/1 [Time: 00:07]
Validating Input:
|#####| 1/1 [Time: 00:07]
Ingesting Batches:
| | 0/2 [Time: 00:00]
Ingesting Batches:
|##5 | 1/2 [Time: 00:08]
Ingesting Batches:
|#####| 2/2 [Time: 00:16]
Ingesting Batches:
|#####| 2/2 [Time: 00:16]
Computing Check:
| | 0/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:00]
Computing Check:
|#####| 1/1 [Time: 00:00]
Total running time of the script: ( 1 minutes 26.071 seconds)