Como uma rede LSTM acelerou a conversão de frequência no LCLS-II

Para montar um experimento de raios X de nível mundial, tudo começa com um laser a funcionar na frequência “errada”.

Antes de se tornar útil, o feixe passa duas vezes por cristais de alta precisão, que elevam a luz do infravermelho para o ultravioleta.

Esse tipo de conversão precisa ser simulado antes que qualquer parâmetro possa ser ajustado.

Até aqui, essa simulação era um gargalo recorrente. Agora, um grupo de físicos substituiu o método por um modelo que executa em milissegundos.

A máquina por trás disso

Jack Hirschman, do SLAC National Accelerator Laboratory (SLAC), liderou uma equipa que contou com investigadores da University of California, Los Angeles (UCLA).

Em conjunto, eles atacaram um dos entraves computacionais mais persistentes da física de lasers moderna. O foco foi o Linac Coherent Light Source II (LCLS-II), uma instalação de raios X de nova geração.

O sistema gera pulsos tão curtos e intensos que conseguem “fotografar” moléculas individuais no meio de uma reação. Para chegar a esse resultado, é necessário cumprir uma cadeia de etapas.

Primeiro, luz laser ultravioleta incide sobre uma superfície metálica e liberta eletrões. Em seguida, esses eletrões são acelerados até à velocidade da luz e depois guiados para um conjunto magnético que os faz emitir raios X.

O pulso ultravioleta que dá início a toda a sequência, porém, nasce como luz infravermelha. Para cumprir a sua função, ele precisa de ser convertido duas vezes.

O processo de conversão

Quando duas ondas de luz entram num cristal especialmente projetado com o ângulo correto, os seus campos elétricos passam a interagir.

Dessa interação surge uma nova onda na frequência combinada - a base da ótica não linear. No LCLS-II, essa etapa recebe o nome de geração de soma de frequências.

Ela ocorre duas vezes em série: o infravermelho transforma-se em verde e, depois, o verde converte-se em ultravioleta. Cada conversão exige um cristal próprio e uma calibração extremamente precisa.

Um estudo sobre a arquitetura do laser da instalação mostra o quanto a qualidade do feixe de eletrões depende da exatidão dessas conversões.

Pequenas irregularidades no pulso UV convertem-se diretamente em degradação do feixe.

O que acontece dentro dos cristais

Antes de os engenheiros ajustarem qualquer parâmetro dessa cadeia, eles precisam de uma simulação do que ocorre no interior dos cristais.

O método convencional resolve uma equação de ondas centenas de vezes ao longo do comprimento do cristal. É um processo preciso, mas lento.

No LCLS-II, esse cálculo não era rápido o suficiente para oferecer retorno em tempo real.

Assim, as equipas executavam simulações offline, analisavam os resultados e faziam ajustes manualmente, num ciclo que não permitia controlo ao vivo.

Uma rede que aprende física

A equipa de Hirschman criou uma alternativa com um tipo de rede neural recorrente chamada rede LSTM (long short-term memory).

Como esse tipo de rede foi concebido para lidar com dados que evoluem em sequência, ele propõe uma forma essencialmente diferente de atacar o problema.

Modelos de aprendizagem profunda assim aprendem padrões temporais durante o treino. Por isso, encaixam-se bem para acompanhar como um pulso de luz muda passo a passo à medida que atravessa um cristal.

Ao trabalhar inteiramente no domínio da frequência, a rede evita a alternância repetida entre representações de frequência e de tempo.

Treino com casos difíceis

O conjunto de treino veio de milhares de simulações do solver convencional, cobrindo uma grande variedade de formatos de pulso.

A equipa incluiu, de propósito, situações difíceis: pulsos com lacunas espectrais e variações de fase intensas - precisamente os cenários em que métodos aproximados mais simples tendem a falhar.

O LSTM não se limitou a prever o pulso principal de saída. Ele acompanhou com precisão os três campos de luz que atravessam o cristal ao mesmo tempo, incluindo as duas ondas de entrada e a onda gerada na saída.

Nenhum substituto (surrogate) anterior para essa classe de interação tinha modelado, de uma só vez, os três campos acoplados em pulsos com formatos tão desafiadores.

Várias vezes mais rápido

Quando executado numa unidade de processamento gráfico (GPU), o modelo substituto conclui cada simulação em milissegundos.

Isso é mais de 250 vezes mais rápido do que o solver convencional. A precisão mantém-se em toda a gama de casos de teste, inclusive nos mais difíceis.

Trabalhos anteriores sobre substitutos de IA na física de lasers já indicavam que acelerações dessa ordem eram possíveis.

O que este estudo acrescenta é a demonstração do método com três campos acoplados num sistema de laser real de um acelerador em operação - um teste mais exigente do que qualquer um dos trabalhos anteriores.

O controlo agora é viável

Com um substituto que responde em milissegundos, a simulação pode ligar-se diretamente ao sistema de controlo do laser.

Dessa forma, engenheiros podem ajustar parâmetros e ver, em tempo real, os resultados previstos antes de realizar qualquer alteração física.

O objetivo mais amplo do grupo é criar gémeos digitais: réplicas completas por simulação de sistemas de laser complexos, atualizadas continuamente em paralelo com o hardware real.

Pela primeira vez, a etapa de conversão em cristais não lineares no centro da cadeia de laser do LCLS-II pode ser modelada com rapidez suficiente para orientar decisões operacionais ao vivo.

Na prática, isso permitiria aos operadores ajustar a ótica de conversão de frequências com retorno preditivo imediato, reduzindo o ciclo de tentativa e erro que hoje depende de computação offline.

A abordagem também pode estender-se a outros sistemas de lasers de alta potência, plataformas de imagem biomédica de pulsos curtos e experiências de fotónica quântica.

O método pode ser aplicado sempre que a física não linear acoplada em cristais precise de ser simulada rapidamente.