Enhancing Classification Performance in Breast Tissue Diagnosis Using TPE-Based Hyperparameter Optimization
K. S. Rangasamy, S. Karthiyayini, Dileep Pulugu, Nabeel Amre, R. Thiagarajan, R. Krishnamoorthy
Abstract
Accurate classification of breast tissue is important for the early medical conditions like carcinoma, fibro-adenoma, and mastopathy. Machine learning models hold promise for automating it, but their performance significantly depends on the selection of the optimal hyperparameters. Traditional optimization techniques, such as grid search or random search, are inefficient and create too much computational cost in handling high-dimensional parameter spaces. In this paper, we present the use of Tree-Structured Parzen Estimator (TPE), a Bayesian probabilistic optimization technique, to optimize hyperparameters for five already wellknown machine learning classifiers: Logistic Regression, SVM, Decision Tree, Random Forest, and k-NN. We adopt this approach to the Breast Tissue Dataset as our benchmark dataset to classify various types of breast tissues. We assess our models with respect to four key performance metrics: Accuracy, Precision, Recall, and F1-Score. The experiments show that TPE optimization leads to consistent improvements in classification accuracy for all the models. For precision, recall, and F1-score, significant gains were observed. Besides, TPE improves model generalization by focusing the search on promising regions of the hyperparameter space, reducing the computational cost compared to traditional methods. This work shows the potential of TPE as a powerful tool for hyperparameter optimization in medical machine learning tasks, providing a more efficient and effective alternative to conventional optimization strategies.