Our project has been successful so far and we have been able to make a lot of progress the last few months. As stated in the previous blog post, we first have to train the model to detect which parts of the MRI images are the Tibia, Femur, and Patella. The Femur bone dataset has been previously labeled and a model has been trained on that dataset already. This makes it easier because we already know the performance percentage and have a good idea of how well formed the model is. We are aiming to have a Tibia model that runs around 96% accuracy or better in tracing the bone. The past few months I and other students have been labeling the Tibia database which consists of 100 cases, the cases are divided into three subfolders. 70% of the data is dedicated to the training folders with the remaining 30% being dedicated to testing and validation folders. The model obviously uses the training data to train the model, and then checks how accurate it is with the validation data. Then the testing data can be used to make predictions and further test the model. There were some complications with the labeled Tibia dataset and some of the cases labeled by me and by other students needed to be relabeled as they weren’t as accurate as they could have been.
We have found that the model doesn’t need the full 70% of the training data to accurately train the model. Both Dr. Shan and I had agreed on a plan to correct the labeling on only a certain percentage of the data (around 30%) and then have the accurate model predict the rest of the cases that we need for the model. With this knowledge we are working on finishing up fitting the model for the Tibia. Once we have completed this model, we plan on moving to the Patella model. The dataset for the Patella is not yet completely labeled and as such we plan on potentially using a trained model to also predict the image masks on the unlabeled Patella data. Once we have all three models complete, we have to combine them into one finished model that has all three bones labeled. This could raise further questions because certain bones start appearing in the MRI image at different times, and we would need to alter the code in ways that could stop any potential errors. Next semester we plan on finishing up the models as quickly as possible so that we can work on getting the main model completed, and move on to detecting the cartilage, one of our final goals.