Switch to unified view

a/README.md b/README.md
...
...
25
25
26
## Self-Supervised Learning
26
## Self-Supervised Learning
27
27
28
Instance-level labels obtained using max-aggregation lack of large sensitivity, since max-aggregation only focus on the most distriminant instances. To alleviate this issue, training an Student model on distilled pseudolabels from the MIL-trained Teacher has shown promising results:
28
Instance-level labels obtained using max-aggregation lack of large sensitivity, since max-aggregation only focus on the most distriminant instances. To alleviate this issue, training an Student model on distilled pseudolabels from the MIL-trained Teacher has shown promising results:
29
29
30
![ssl](https://github.com/jusiro/mil_histology/blob/main/images/student_method.png)
30
![ssl](https://github.com/jusiro/mil_histology/blob/main/images/student_method.png?raw=true)
31
31
32
```
32
```
33
python train_student.py --experiment_name instance_max
33
python train_student.py --experiment_name instance_max
34
```
34
```
35
35
...
...
38
**J. Silva-Rodríguez, A. Colomer, J. Dolz and V. Naranjo, "Self-Learning for Weakly Supervised Gleason Grading of Local Patterns," in IEEE Journal of Biomedical and Health Informatics, vol. 25, no. 8, pp. 3094-3104, 2021.** [(paper)](https://ieeexplore.ieee.org/abstract/document/9361085)[(arXiv)](https://arxiv.org/abs/2105.10420)
38
**J. Silva-Rodríguez, A. Colomer, J. Dolz and V. Naranjo, "Self-Learning for Weakly Supervised Gleason Grading of Local Patterns," in IEEE Journal of Biomedical and Health Informatics, vol. 25, no. 8, pp. 3094-3104, 2021.** [(paper)](https://ieeexplore.ieee.org/abstract/document/9361085)[(arXiv)](https://arxiv.org/abs/2105.10420)
39
39
40
## Proportion Constraints
40
## Proportion Constraints
41
Weakly supervised learning methods usually require large amounts od data to perform properly.In this project, we propose to introduce proportion priors per WSI as additional, weak information, to train realiable deep learning models. Concretely, we take advantadge of the Gleason scoring system, which defines a primary and secondary class per bag based on tissue proportions. Thus, we use inequality constraints to ensure that the percentage of positive predictions for the primary class is larger than the one for the secondary grade. Our formulation is flexible, and might deal with other problems, where more constraints are known regarding relative class proportions. You can train the proposed model using the following code.
41
Weakly supervised learning methods usually require large amounts od data to perform properly.In this project, we propose to introduce proportion priors per WSI as additional, weak information, to train realiable deep learning models. Concretely, we take advantadge of the Gleason scoring system, which defines a primary and secondary class per bag based on tissue proportions. Thus, we use inequality constraints to ensure that the percentage of positive predictions for the primary class is larger than the one for the secondary grade. Our formulation is flexible, and might deal with other problems, where more constraints are known regarding relative class proportions. You can train the proposed model using the following code.
42
42
43
![pc](https://github.com/jusiro/mil_histology/blob/main/images/proportion_method.png)
43
![pc](https://github.com/jusiro/mil_histology/blob/main/images/proportion_method.png?raw=true)
44
44
45
```
45
```
46
python main.py --experiment_name instance_max_Constrained --aggregation max --mode instance --pMIL True --alpha_ce 1 --alpha_ic 0.1 --alpha_pc 1 --t_ic 15 --t_pc 5 --scheduler True --early_stopping True --criterion z --epochs 100
46
python main.py --experiment_name instance_max_Constrained --aggregation max --mode instance --pMIL True --alpha_ce 1 --alpha_ic 0.1 --alpha_pc 1 --t_ic 15 --t_pc 5 --scheduler True --early_stopping True --criterion z --epochs 100
47
```
47
```
48
48
...
...
57
Finally, you can produce visualization of instance-level predicitons through the following code:
57
Finally, you can produce visualization of instance-level predicitons through the following code:
58
58
59
```
59
```
60
python produce_visualizations.py --experiment_name instance_max_Constrained
60
python produce_visualizations.py --experiment_name instance_max_Constrained
61
```
61
```
62
![visualizations](https://github.com/jusiro/mil_histology/blob/main/images/visualzations.png)
62
![visualizations](https://github.com/jusiro/mil_histology/blob/main/images/visualzations.png?raw=true)
63
63
64
## Contact
64
## Contact
65
For further questions or details, please directly reach out to Julio Silva-Rodríguez
65
For further questions or details, please directly reach out to Julio Silva-Rodríguez
66
(jusiro95@gmail.com)
66
(jusiro95@gmail.com)