Cómo enseñar a un programa a dibujar la zona de irradiación ✛
Cuando un médico trata el cáncer de estómago con radioterapia, debe delimitar con precisión la zona a irradiar. Esa zona no se ve como un órgano independiente, así que el médico tiene que tener en mente la ubicación de todos los órganos vecinos. El programa nnU-Net aprende a hacer lo mismo, pero primero hay que mostrarle ejemplos.
Los investigadores le dieron al programa un mapa con todos los órganos de la cavidad abdominal marcados. Es como un plano de calles con los edificios etiquetados: el programa ve dónde está el estómago, dónde está el intestino y no invade territorio ajeno. Además, lo reentrenaron con los casos más difíciles, aquellos en los que el propio programa no estaba seguro de la respuesta. Así aprende de sus errores.
Tras este entrenamiento, el programa dibuja el contorno casi tan bien como los médicos experimentados: su resultado es 0.87, mientras que el de los médicos varía en torno a 0.89. Ahora el médico puede confiar al programa un borrador y revisar solo los casos dudosos. Esto ahorra tiempo y ayuda a tratar a los pacientes.
El entrenamiento de un solo modelo tomó unas 22 horas en una GPU NVIDIA Quadro 8000, mientras que el etiquetado de los puntos de referencia anatómicos para una TC tomó solo 6 minutos. Añadir este mapa de 6 minutos ahorró cuatro rondas de aprendizaje activo, cada una de las cuales requería la revisión manual de los contornos por dos oncólogos.