Hello Andy,
I understand that you are facing issues while training "3-D Brain Tumor Segmentation Using Deep Learning" example on your own data. This issue can occur due to various reasons, such as incorrect data alignment, inadequate data augmentation, inappropriate hyperparameter settings, suboptimal model architecture, insufficient training duration, or class imbalance.
You can follow the below steps to resolve this issue:
- Data-related Issues: Verify data alignment and preprocessing to ensure input and labels are correctly aligned. Check for anomalies or inconsistencies in the data that might affect training.
- Data Augmentation: Adjust the intensity or types of transformations applied during augmentation. Experiment with different augmentation strategies to find the right balance.
- Hyperparameter Tuning: Tune the learning rate, batch size, and dropout rate to optimize the model's performance. Reduce the learning rate or adjust the drop factor for slower rates of decay.
- Model Architecture: Modify the architecture by adding or removing layers, adjusting the number of filters, or trying different layer types. Ensure the complexity and depth of the network are suitable for your dataset.
- Training Duration: Increase the number of epochs or train for a longer duration to allow the model to converge effectively.
- Class Imbalance: Apply class balancing techniques such as class weighting or oversampling/undersampling to address any significant class imbalance.
You can refer to the below documentation to learn more about Data Alignment and Preprocessing, Data Augmentation and Class Imbalance in MATLAB.
