TauPolaris: reconstructing tau lepton polarimetric vectors with conditional normalizing flows

Resumo Técnico: TauPolaris

Problema
A orientação de spin dos léptons tau ($\tau$) é um observável crítico para sondar as propriedades fundamentais do Modelo Padrão e além, incluindo buscas por violação de CP em decaimentos do bóson de Higgs ($H \to \tau\tau$), testes de emaranhamento quântico e medições de operadores efetivos. O observável ótimo para determinar o spin do $\tau$ é o vetor polarimétrico, que aponta na direção mais provável do spin do $\tau$. No entanto, a reconstrução deste vetor é excepcionalmente desafiadora porque os decaimentos do $\tau$ invariavelmente produzem neutrinos que escapam da detecção.

Os métodos experimentais atuais dependem de construções aproximadas para observáveis sensíveis ao spin (por exemplo, usando parâmetros de impacto ou planos de decaimento) porque os momentos dos neutrinos são desconhecidos. Essas aproximações limitam a precisão experimental e são difíceis de transferir entre diferentes processos. Além disso, abordagens padrão de aprendizado de máquina que realizam regressão de ponto (minimizando o erro quadrático médio) frequentemente falham em capturar as correlações complexas e multimodais entre os componentes do momento do neutrino. Quando a cinemática permite duas soluções distintas, um modelo de regressão de ponto tende a prever um valor médio entre elas, enviesando a direção do neutrino em direção ao momento do $\tau$ visível e degradando as correlações angulares essenciais para as medições de spin.

Metodologia
Os autores apresentam o TauPolaris, uma ferramenta que reconstrói vetores polarimétricos de tau estimando os momentos dos neutrinos não detectados usando fluxos de normalização condicionais (conditional normalizing flows). Diferente das redes de regressão de ponto, este método modela a densidade de probabilidade condicional completa, $p(\nu | c)$, da cinemática do neutrino dado o contexto reconstruído do evento $c$.

Principais características arquitetônicas e metodológicas incluem:

• Fluxo de Normalização Condicional: O modelo mapeia a cinemática do neutrino $\nu$ para uma variável latente $z$ que segue uma distribuição Gaussiana padrão através de uma série de transformações invertíveis. Isso permite tanto a amostragem da densidade completa (para reproduzir distribuições de nível de gerador) quanto a determinação da estimativa de Máxima Verossimilhança a Posteriori (MAP) (a configuração mais provável).
• Sistema de Coordenadas: Os momentos dos neutrinos são parametrizados em uma base ortonormal $\{n, r, k\}$ definida em relação ao momento do $\tau$ visível. Isso alinha a saída do modelo com a base usada para expressar correlações de spin, melhorando a resolução.
• Características de Entrada: O modelo utiliza uma rede de condicionamento baseada em transformer para processar as entradas do evento, incluindo os quatro-momentos dos produtos de decaimento visíveis (píons carregados, léptons e faixas de $\pi^0$ reconstruídas), parâmetros de impacto, posições de vértices secundários e momento transversal ausente.
• Estimativa MAP Baseada em Gradiente: Para obter a configuração de neutrino mais provável para um evento específico sem depender de amostragem estocástica, os autores empregam uma rotina de maximização baseada em gradiente no espaço latente. Esta abordagem é mais estável e confiável do que a maximização direta sobre os componentes do neutrino.
• Treinamento: Os modelos são treinados em dados simulados de colisões próton-próton do LHC (Pythia 8.312) com efeitos de detector modelados via Delphes 3.5 e borramento (smearing) customizado. Modelos separados são treinados para os canais totalmente hadrônico ($\tau_h\tau_h$) e semilepônico ($\tau_\ell\tau_h$).

Principais Contribuições

1. Modelagem de Densidade Completa: Ao modelar a densidade conjunta dos componentes do momento em vez de estimativas de ponto independentes, o TauPolaris preserva as correlações físicas entre os componentes do momento. Isso evita o "viés de colinearidade" observado em regressores de erro quadrático médio, onde o neutrino previsto é artificialmente alinhado com o $\tau$ visível.
2. Quantificação de Incerteza: O método fornece não apenas uma estimativa de ponto (MAP), mas também uma estimativa de incerteza por evento derivada da dispersão da densidade amostrada.
3. Extensão a Todas as Topologias: A abordagem estende o uso de vetores polarimétricos a todas as topologias de decaimento de $\tau$, incluindo aquelas anteriormente difíceis de analisar (ex: $3\pi^\pm 1\pi^0$ e decaimentos leptônicos), ao reconstruir explicitamente a cinemática do neutrino.
4. Novas Variáveis de Supressão de Fundo: Os vetores polarimétricos reconstruídos são usados para definir novos observáveis que exploram correlações de spin para distinguir eventos $H \to \tau\tau$ de fundos $Z \to \tau\tau$ sem assumir um ângulo de mistura de CP específico.

Resultos
O desempenho do TauPolaris foi avaliado em três aplicações de referência usando dados simulados do HL-LHC (High-Luminosity LHC):

• Emaranhamento Quântico: Os autores demonstram que a presença de emaranhamento quântico em decaimentos $H \to \tau\tau$ pode ser distinguida de um estado separável (não emaranhado) com uma significância de pelo menos 4.3$\sigma$ no HL-LHC. Isso é alcançado reconstruindo os elementos da matriz de densidade de spin ($C_{ij}$) e calculando a concorrência ($C$) e o testemunho de não-localidade de Bell ($m_{12}$).
• Sensibilidade à Violação de CP: O método melhora a sensibilidade à violação de CP em decaimentos $H \to \tau\tau$ em 18% comparado a métodos anteriores de reconstrução aproximada. Isso é quantificado pela assimetria nas distribuições do ângulo de acoplanaridade ($\phi_{CP}$). A melhoria é mais significativa nos modos de decaimento envolvendo três hádrons carregados (ex: $3\pi^\pm 0\pi^0$), onde a assimetria aumenta em 88% em relação aos métodos aproximados.
• Supressão de Fundo: Novas variáveis construídas a partir dos vetores polarimétricos (especificamente combinações de $\cos \theta_n \cos \theta_n$ e $\cos \theta_r \cos \theta_r$) separam efetivamente eventos de Higgs spin-0 de eventos de bóson $Z$ spin-1. Essas variáveis são independentes do estado de CP do Higgs, fornecendo um controle robusto para a supressão de fundo.
• Resolução: A resolução dos observáveis de spin reconstruídos (definida como o intervalo interquartil do erro de previsão) melhora em aproximadamente 40% comparado a um regressor transformer isolado treinado com erro quadrático médio.

Significância
O artigo afirma que o TauPolaris representa um avanço significativo na polarimetria de léptons tau ao ir além de construções cinemáticas aproximadas para uma reconstrução rigorosa de densidade dos momentos dos neutrinos. Ao modelar com precisão a densidade condicional da cinemática do neutrino, a ferramenta recupera os observáveis de análise de spin ótimos com maior precisão e fornece estimativas de incerteza por evento. Essa capacidade permite testes mais sensíveis da física fundamental, incluindo a primeira demonstração robusta de distinção de emaranhamento quântico de correlações clássicas em $H \to \tau\tau$ no HL-LHC e uma redução substancial na incerteza das medições de violação de CP. Os autores observam que a capacidade de estimar incertezas de reconstrução polarimétrica por evento oferece potencial adicional para otimizar seleções de eventos em análises futuras.