A tool to automatically parametrize the effect of EFT coefficients on arbitrary observables.
- Initial setup
- Setup models
- Setup Rivet routines
- Process-specific workflow
- Standalone reweighting
- Known limitations
The assumption is that the cross section for a bin i can be expressed as the sum
where σ_int is the leading term in the EFT expansion (going as 1/Λ^2), and accounts for the interference with the SM amplitude. The SM-independent term is labelled as σ_BSM (going as 1/Λ^4).
Dividing through by the SM cross section, σ_SM, provides a scaling function for each bin
which parametrizes deviations in the cross section in terms of the c_j parameters. The EFT2Obs tool determines the coefficients A_j and B_jk by generating events in which matrix element weights for different c_j values are embedded.
Note that this workflow has only been tested on lxplus so far, and for CMS users please ensure the CMSSW environment is not set when running scripts in EFT2Obs.
Follow these steps to set up and install the relevant software packages.
Clone the EFT2Obs repository and source the env.sh
script which sets a number of useful environment variables:
git clone https://github.com/ajgilbert/EFT2Obs.git
cd EFT2Obs
source env.sh
The env.sh
script should be sourced at the start of each new session.
Then run the following scripts to download and install LHAPDF, Madgraph_aMC@NLO, Pythia and Rivet. Note this may take some time to complete.
./scripts/setup_mg5.sh
./scripts/setup_rivet.sh
The first script installs LHAPDF and Magraph, then applies patches to the Madgraph-Pythia interface that are needed to make valid HepMC output with stored event weights and to perform a transformation of these weights that is described in more detail below.
The Madgraph-compatible models you wish to study should be installed next. Scripts are provided to do this for two commonly used models: the Higgs Effective Lagrangian (HEL) and SMEFTsim. Other models can be added following their respective instructions.
./scripts/setup_model_HEL.sh
./scripts/setup_model_SMEFTsim.sh
Note: a patch is provided for the HEL model that will fix the mass of the Z boson, otherwise by default this will vary as a function of the EFT parameters: patch MG5_aMC_v2_6_7/models/HEL_UFO/parameters.py setup/HEL_fix_MZ__parameters.py.patch
The Rivet routines which define the fiducial regions and obserbables of interest should be placed in the EFT2Obs/RivetPlugins
directory. A routine for the Higgs STXS is already provided: HiggsTemplateCrossSections
which contain the stage 1.2 definitions. Whenever new routines are added or existing ones modified they should be recompiled with:
./scripts/setup_rivet_plugins.sh
The main steps for defining a process, generating events and extracting scaling factors for observables are outlined below:
In this section for clarity we will follow a specific example, ZH production in the HEL model with the STXS Rivet routine, but the steps are generic and it should be straightforward to run with other processes or models.
It is also possible to generate events at NLO in QCD and apply reweighting with an NLO-capable UFO model. In some places an alternative script must be used, and in others it is sufficient to add the --nlo
flag to the usual script.
Each process+model combination we want to study first needs a dedicated sub-directory in EFT2Obs/cards
. In this case we will use the pre-existing one for zh-HEL
. Here we add a proc_card.dat
specifying the model to import and the LO process we wish to generate:
import model HEL_UFO
# Definitions
define ell+ = e+ mu+ ta+
define ell- = e- mu- ta-
define j2 = g u c d s u~ c~ d~ s~
generate p p > h ell+ ell- / j2 ell+ ell- vl vl~ a h w+ w- NP<=1 @0
add process p p > h vl vl~ / j2 ell+ ell- vl vl~ a h w+ w- NP<=1 @1
output zh-HEL
Notes on the param_card.dat
format:
- A restriction on the number of new physics vertices to be less than one should be applied. In many models this is achieved with adding
NP<=1
at the end of each process, but check the model-specific documentation for guidance. See the Known limitations section below for more details. - The argument to the
output
command at the end must match thecards
sub-directory name.
To initialise this process in Madgraph run
./scripts/setup_process.sh zh-HEL
which creates the directory MG5_aMC_v2_6_7/zh-HEL
.
For an NLO process (with [QCD]
in the process definition), use the setup_process_NLO.sh
script instead.
There are four further configuration cards that we need to specify: the run_card.dat
, pythia8_card.dat
, param_card.dat
and reweight_card.dat
. For NLO generation the pythia8_card.dat
card is replaced by shower_card.dat
. For the first two we can start from the default cards Madgraph created in the MG5_aMC_v2_6_7/zh-HEL/Cards
directory. If these files do not already exist in our cards/zh-HEL
directory then they will have been copied there in the setup_process.sh
step above. If necessary edit these cards to set the desired values for the generation or showering parameters. In this example the cards have already been configured in the repository. Note for CMS users: to emulate the Pythia configuration in official CMS sample production the lines in resources/pythia8/cms_default_2018.dat
can be added to the pythia8_card.cat
.
For NLO generation some care must be taken with the run_card.dat
:
- The precision for the integration grids is controlled by
req_acc
. We recommend setting this to0.001
, as recommended by the MG authors are typically used in official CMS production. - Make sure
parton_shower
is set toPYTHIA8
. - To use Madgraph's NLO-correct reweighting, the
store_rwgt_info
flag must be set to true. For this information to be stored, one ofreweight_scale
orreweight_PDF
must be set to true as well. We recommendreweight_scale
, and setting bothrw_rscale
andrw_fscale
to1.0
. - Check that
ickkw
is set appropriately if you will use FxFx merging.
To define param_card.dat
and create reweight_card.dat
we first need to make an EFT2Obs-specific configuration file that will keep track of which (subset of) model parameters we want to study and what values they should be set to in the reweighting procedure. The initial config is generated with
python scripts/make_config.py -p zh-HEL -o config_HEL_STXS.json \
--pars newcoup:4,5,6,7,8,9,10,11,12 --def-val 0.01 --def-sm 0.0 --def-gen 0.0
where:
-p
gives our process label-o
the output name for the config file--pars
specifies which parameters we are interested in--def-val
is a default nominal non-SM value to be used in the reweighting--def-sm
is the default value of the parameters that corresponds to the SM expectation--def-gen
the default parameter value used to generate the events
The --pars
option supports multiple arguments of the form [BLOCK]:[ID1],[ID2],...[IDN]
. The block name and numbers correspond to those in the param_card.dat
that Madgraph produces in the MG5_aMC_v2_6_7/zh-HEL/Cards
directory. The resulting file looks like:
{
"blocks": ["newcoup"],
"inactive": {
"default_val": 0.0,
"parameters": []
},
"parameter_defaults": {
"block": "newcoup", "gen": 0.0, "sm": 0.0, "val": 0.01
},
"parameters": [
{"index": 4, "name": "cu"},
{"index": 5, "name": "cd"},
{"index": 6, "name": "cl"},
{"index": 7, "name": "cww"},
{"index": 8, "name": "cb"},
{"index": 9, "name": "chw"},
{"index": 10, "name": "chb"},
{"index": 11, "name": "ca"},
{"index": 12, "name": "cg"}
]
}
All properties in the parameter_defaults
block are assumed to apply to each parameter specified in the parameters
list, however these can be modified on a parameter-by-parameter basis by setting, e.g. {"index": 4, "name": "cu", "val": 0.02}
.
Note that all parameters in the specified blocks not explicitly listed here will be set to default_val
in the inactive
part of the config. This will usually correspond to the SM value for the parameters. If some parameters should take a different value then this can be set with the argument --set-inactive [BLOCK]:[ID1]=[VAL1],[ID2]=[VAL2] ...
.
Next we make the param_card.dat
that will be used for the initial event generation:
python scripts/make_param_card.py -p zh-HEL -c config_HEL_STXS.json -o cards/zh-HEL/param_card.dat
The script take the default card in MG5_aMC_v2_6_7/zh-HEL/Cards
and will report which parameter values are changed.
The reweight card specifies a set of parameter points that should be evaluated in the matrix element reweighting step. It is generated using the config with:
python scripts/make_reweight_card.py config_HEL_STXS.json cards/zh-HEL/reweight_card.dat
For NLO processes, an extra line should be added to the reweight_card.dat
to select the NLO-correct reweighting mode (if desired):
python scripts/make_reweight_card.py config_HEL_STXS.json cards/zh-HEL/reweight_card.dat --prepend 'change mode NLO'
To make the event generation more efficient and easier to run in parallel we first produce a gridpack for the process:
./scripts/make_gridpack.sh zh-HEL # for LO
./scripts/make_gridpack_NLO.sh zh-HEL # for NLO
Once complete the gridpack gridpack_zh-HEL.tar.gz
will be copied to the main directory. The script can also produce a standalone version of the matrix-element code. This can be useful for reweighting events outside of the EFT2Obs tool. To produce this, add an additional flag:
./scripts/make_gridpack.sh zh-HEL 1
An addtional file, rw_module_zh-HEL.tar.gz
is also produced. See the section below on standalone reweighting for more information. Note that standalone reweighting with NLO matrix elements is not currently possible.
Now everything is set up we can proceed to the event generation.
This is handled by scripts/run_gridpack.py
, which runs through the LHE file generation with Madgraph, the EFT model reweighting, showering with Pythia and finally processing with Rivet. The output is a yoda file containing all the Rivet routine histograms. We exploit the feature that in Rivet 3.X a copy of each histogram is saved for every weight. The following command generates 500 events:
export HIGGSPRODMODE=ZH
python scripts/run_gridpack.py --gridpack gridpack_zh-HEL.tar.gz -s 1 -e 500 \
-p HiggsTemplateCrossSectionsStage1,HiggsTemplateCrossSections \
-o test-zh [--nlo]
where the full set of options is:
--gridpack [file]
relative or absolute path to the gridpack--nlo
should be set if the input is an NLO gridpack-s X
the RNG seed which will be passed to Madgraph. This is also used to index the output files from the job, e.g.Rivet_[X].yoda
.-e N
number of events per job, in this case 500.-p [RivetRoutine1],[RivetRoutine2],...
comma separated list of Rivet routines to run-o [dir]
the output directory for the yoda files, will be created if it does not exist--save-lhe [path]
save the intermediate LHE files (containing the event weights) to the relative or absolute directory given by[path]
. The output file will be of the form[path]/events_[X].lhe.gz
where[X]
is the RNG seed.--save-hepmc [path]
save the intermediate HepMC files to the relative or absolute directory given by[path]
. The output file will be of the form[path]/events_[X].hepmc.gz
where[X]
is the RNG seed.--load-lhe [path]
load the existing LHE file from the relative or absolute directory given by[path]
. The input file must be of the form[path]/events_[X].lhe.gz
where[X]
is the RNG seed. All prior steps are skipped.--load-hepmc [path]
load the existing HepMC file from the relative or absolute directory given by[path]
. The input file must be of the form[path]/events_[X].hepmc.gz
where[X]
is the RNG seed. All prior steps are skipped. Note that it does not make sense to set both--load-lhe
and--load-hepmc
.--rivet-ignore-beams
sets the--ignore-beams
option when running Rivet, useful for cases where only a particle decay is simulated, instead of a full collision.--no-cleanup
prevents the local working directory (namedgridpack_run_[X]
) from being deleted at the end, can be useful for debugging.--to-step [lhe,shower,rivet]
stop after the given step, no further processing is performed. Useful if, for example, you only want to generate the LHE files and not run the rivet routine.
Since it is usually not feasible to generate the desired number of events in a single job, a wrapper script scripts/launch_jobs.py
is provided which can automate running a set of jobs in parallel, each with a different RNG seed so that the events are statistically independent. This script passes through all command line options to run_gridpack.py
, but adds several extra options to control how the jobs run:
-j N
the number of jobs to run-s X
the initial RNG seed for the first job. Subsequent jobs i useX + i
as the seed.--job-mode
supported options are interactive or submission to batch farms, e.g. condor--parallel X
number of parallel jobs to run in interactive mode--env X1=Y1,X2=Y2,..
any environment variables that should be set in the job. In this example the STXS Rivet routines require the Higgs production mode to be specified.
The following command runs four jobs, each generating 500 events:
python scripts/launch_jobs.py --gridpack gridpack_zh-HEL.tar.gz -j 4 -s 1 -e 500 \
-p HiggsTemplateCrossSectionsStage1,HiggsTemplateCrossSections \
-o test-zh --job-mode interactive --parallel 4 --env HIGGSPRODMODE=ZH
To submit jobs to htcondor at CERN the --job-mode interactive --parallel 4
options can be replaced by --job-mode condor --task-name stxs --sub-opts '+JobFlavour = "longlunch"'
. Anything specified in --sub-opts
will be added to the condor submit file.
Note that during the workflow the event weights that are added by Madgraph are transformed before they are utilised by Rivet. This is done to isolate the separate terms that will appear in the final parametrization, and to more easily keep track of their statistical uncertainties. The table below illustrates the weight transformation for a simple example with two model parameters, c_1 and c_2. For each weighting point, labelled W_i, the expression for the default and transformed weights are given as a function of the D_1 and D_2 constants, which take the values specified in the config file.
In this final step we first merge the yoda output files:
yodamerge -o test-zh/Rivet.yoda test-zh/Rivet_*
and then use the script get_scaling.py
to extract the A_j and B_jk coefficents and their statistical uncertainties:
python scripts/get_scaling.py -c config_HEL_STXS.json [--nlo] \
-i test-zh/Rivet.yoda --hist "/HiggsTemplateCrossSections/pT_V" \
--save json,txt,tex --translate-tex resources/translate_tex.json \
--rebin 0,10,20,30,40,50,60,70,80,90,100,120,140,160,180,200,250,300,350,400
where the options are
-c
the config file to use--nlo
should be set if the output came from NLO reweighting-i
the yoda file input--hist
the histogram in the yoda file to parametrize, note that the leading "/" is important!--save [json,txt,tex]
a comma separated list of the output formats to save in--translate-tex
a json dictionary file that converts the parameter names to latex format - only needed whentex
is chosen in--save
--rebin
optionally re-bin the histogram first by providing a list of the new bin edges
The terms calculated for each bin are printed to the screen, e.g.
-----------------------------------------------------------------
Bin 0 numEntries: 4182 mean: 1.25e-10 stderr: 1.28e-14
-----------------------------------------------------------------
Term | Val | Uncert | Rel. uncert.
-----------------------------------------------------------------
cww | 23.8014 | 0.0676 | 0.0028
cb | 6.5793 | 0.0269 | 0.0041
chw | 3.9987 | 0.0165 | 0.0041
chb | 1.1975 | 0.0049 | 0.0041
ca | 4.5721 | 0.0377 | 0.0083
cww^2 | 146.3984 | 1.3780 | 0.0094
cb^2 | 11.5790 | 0.1360 | 0.0117
chw^2 | 4.4363 | 0.1007 | 0.0227
chb^2 | 0.3978 | 0.0090 | 0.0227
ca^2 | 6.7260 | 0.1295 | 0.0193
cww * cb | 80.3150 | 0.8199 | 0.0102
cww * chw | 48.3712 | 0.4014 | 0.0083
cww * chb | 14.4853 | 0.1202 | 0.0083
cww * ca | 55.4565 | 0.7574 | 0.0137
cb * chw | 13.3996 | 0.1229 | 0.0092
cb * chb | 4.0127 | 0.0368 | 0.0092
cb * ca | 17.0145 | 0.2641 | 0.0155
chw * chb | 2.6570 | 0.0603 | 0.0227
chw * ca | 9.1324 | 0.0968 | 0.0106
chb * ca | 2.7348 | 0.0290 | 0.0106
The output in json
format is intended for further processing. It contains information on the histogram edges as contents as well as all the scaling terms and their statistical uncertainties. The txt
and tex
formats provide a more readable format:
HiggsTemplateCrossSections_pT_V.txt
0-10:1 + 23.801 * cww + 6.579 * cb + 3.999 * chw + 1.197 * chb + 4.572 * ca + 146.398 * cww * cww + 11.579 * cb * cb + 4.436 * chw * chw + 0.398 * chb * chb + 6.726 * ca * ca + 80.315 * cww * cb + 48.371 * cww * chw + 14.485 * cww * chb + 55.456 * cww * ca + 13.400 * cb * chw + 4.013 * cb * chb + 17.014 * cb * ca + 2.657 * chw * chb + 9.132 * chw * ca + 2.735 * chb * ca
10-20:1 + 23.866 * cww + 6.639 * cb + 4.118 * chw + 1.233 * chb + 4.623 * ca + 146.841 * cww * cww + 11.738 * cb * cb + 4.828 * chw * chw + 0.433 * chb * chb + 6.825 * ca * ca + 80.973 * cww * cb + 50.028 * cww * chw + 14.982 * cww * chb + 55.884 * cww * ca + 13.938 * cb * chw + 4.174 * cb * chb + 17.244 * cb * ca + 2.891 * chw * chb + 9.443 * chw * ca + 2.828 * chb * ca
HiggsTemplateCrossSections_pT_V.tex
A script is available for plotting the histogram overlaying the expectation for arbitrary values of the parameters. For this ROOT (and the PyROOT interface) must be available. The script relies only on the json output file from the above step, for example:
python scripts/makePlot.py --hist HiggsTemplateCrossSections_pT_V.json -c config_HEL_STXS.json --x-title "p_{T}(Z) [GeV]" --title-left "qq #rightarrow Hl^{+}l^{-}" --title-right "HEL UFO" --ratio 0.9,1.95 --draw chw=0.005:4 cww=0.005:2 cb=0.005:8 ca=0.05:12 --show-unc --y-min 1E-9 --translate resources/translate_root.json
where the arguments to --draw
are of the format PAR1=X1,PAR2=X2,..:COLOR
. This example gives the output:
The matrix element library madgraph generates can be exported and used standalone with a python interface. EFT2Obs includes a wrapper python module, scripts/standalone_reweight.py
that makes it straightforward to run the full set of reweighting points on a given event. This module has no dependencies on other EFT2Obs code so can be run from any location.
The steps to making a complete standalone directory are:
- The standalone module can be produced in one of two ways. The first is when running
make_gridpack.sh
, we can add an extra flag to also export the standalone matrix element library:
./scripts/make_gridpack.sh zh-HEL 1
# An addtional file, rw_module_zh-HEL.tar.gz, is created
Alternatively, if we are not interested in actually generating events, we can skip the integration and just generate and compile the matrix element code:
./scripts/setup_process_standalone.sh zh-HEL
This generates a special process directory with the -standalone
postfix in the Madgraph directory.
- Run the script
make_standalone.py
. This creates a directory, specified by-o
which will contain: the matix library from madgraph, a copy of the EFT2Obs config file (specified with-c
), and a set of full param_cards, one for each reweighting point.
If using the gridpack output in 1):
python scripts/make_standalone.py -p zh-HEL -o rw_zh-HEL -c config_HEL_STXS.json --rw-module rw_module_zh-HEL.tar.gz
If using setup_process_standalone.sh
without making a gridpack:
python scripts/make_standalone.py -p zh-HEL -o rw_zh-HEL -c config_HEL_STXS.json --rw-dir MG5_aMC_v2_6_7/zh-HEL-standalone
- The output directory can then be used in conjunction with the
standalone_reweight.py
module:
from standalone_reweight import *
rw = StandaloneReweight('rw_zh-HEL')
# Now specify the list of ingoing and outgoing particles and other properties.
# NB: all lists must be the same length!
parts = [
# Incoming parton 1: [E, px, py, pz],
# Incoming parton 2: [E, px, py, pz],
# Outgoing particle 1: [E, px, py, pz],
# ...
#Outgoing particle N: [E, px, py, pz]
]
pdgs = [] # List PDGs corresponding to the above particles [int]
helicities = [] # List of helicities corresponding to the above particles [int]
status = [] # # List of status codes, should be -1 for incoming, +1 for outgoing [int]
alphas = 0.137 # event-specific value of alphas
use_helicity = True # Set to False to sum over helicity states, e.g. if helicity information not available
# Returns a list of weights in the same order specified by the EFT2Obs reweight_card.dat
weights = rw.ComputeWeights(parts, pdgs, helicities, status, alphas, use_helicity)
The ComputeWeights function will print an error message if the particle configuration is not defined in the matrix element library. These raw weights are not always useful for further processing, so a transformation function is provided that isolates the different linear and quadratic components the same way in normal EFT2Obs usage:
transformed_weights = rw.TransformWeights(weights)
Given the transformed weights, a new weight can be calculated for arbitrary coefficient values:
new_weight = rw.CalculateWeight(transformed_weights, cW=0.1, cHW=0.1)
The following limitations currently apply. Links to GitHub issues indicate which are being actively worked on. Other issues or feature requests should also be reported there.
- Processes are limited to one new-physics vertex (i.e.
NP <= 1
syntax required in the process definition). - Some incompatibilities with the CMSSW environment have been reported. For now this should not be set when running EFT2Obs.