Dynamic data balancing strategy-based Xception-dual-channel LSTM model for laparoscopic cholecystectomy phase recognition
摘要
To enhance the temporal feature learning capability of the laparoscopic cholecystectomy phase recognition model and address the class imbalance issue in the training data, this paper proposes an Xception-dual-channel LSTM fusion model based on a dynamic data balancing strategy.
MethodsThe model dynamically adjusts the undersampling rate for each surgical phase, extracting short video clips from the original data as training samples to balance the data distribution and mitigate biased learning. The Xception model, utilizing depthwise separable convolutions, extracts fundamental visual features frame by frame, which are then passed to a dual-channel LSTM network. This network is composed of a temporal mapping bidirectional LSTM structure and a sequence embedding LSTM structure, both working in parallel. The dual-channel LSTM network models the temporal dependencies between adjacent frames, capturing the contextual temporal information to perceive the dynamic feature changes of the surgical phases. Finally, the surgical phase is determined by combining the prediction scores from both channels.
ResultsExperimental evaluation on the public dataset Cholec80 demonstrates that the proposed model outperforms traditional single-channel LSTM models. Moreover, compared to the model without the dynamic data balancing strategy, the F1-scores for all surgical phases have been improved.
ConclusionThe experimental results validate the effectiveness of this strategy in extracting temporal feature information, alleviating the data class imbalance issue, and enhancing the overall detection performance of the model.