1 Introduction

The search for dark matter and the investigation of the fundamental nature of neutrinos are two outstanding endeavours in contemporary physics. The dual-phase xenon time projection chambers (TPCs), led by the XENON1T experiment, has achieved to date the most stringent upper limits on spin-independent [1] and spin-dependent neutron [2] interactions for WIMPs with mass above \(6\, {\hbox {GeV/c}}^2\), as well as for sub-GeV dark matter particles [3]. XENON1T uses xenon containing \(^{136} \hbox {Xe}\) with isotopic abundance of 8.49%, it can therefore also search for the neutrinoless double-beta decay (\(0\nu \beta \beta \)) at its Q value, \(Q_{\beta \beta } = (2457.83 \pm 0.37)\, \hbox {keV}\) [4]. A detection of \(0\nu \beta \beta \) would establish the Majorana nature of neutrinos and demonstrate lepton number violation by two units. The experimental signature of \(0\nu \beta \beta \) is a mono-energetic peak at \(Q_{\beta \beta }\), at the falling end of the continuous energy spectrum of the two-neutrino double beta decay (\(2\nu \beta \beta \)) standard model process. The \(0\nu \beta \beta \) half-life sensitivity depends on the total detection efficiency, \(\epsilon \), the isotopic abundance, \(n_{136}\), the atomic mass number \(m_A\) of \(^{136} \hbox {Xe}\), and the total exposure \(M\cdot t\), where M is the fiducial mass, and t is the livetime of the measurement. In the absence of signal events, in an energy interval \({\varDelta }E\) around \(Q_{\beta \beta }\), the 90% CL limit on the half-life can be expressed as

$$\begin{aligned} T_{1/2}^{0\nu } > \frac{\ln 2}{1.64} \frac{N_A}{m_A} \cdot \epsilon \cdot n_{136} \cdot \frac{M \cdot t}{\sqrt{n_B}} \propto \sqrt{\frac{M \cdot t}{B \cdot {\varDelta }E}}\text { ,} \end{aligned}$$
(1)

where \(N_A\) is Avogadro’s number, \(n_B\) is the number of expected background events and B is the background rate in the energy interval [5]. A good energy resolution is fundamental to minimize the region \({\varDelta }E\), thus enhancing the experimental sensitivity. This paper describes several improvements to the signal reconstruction algorithms for XENON1T, leading to excellent energy linearity and resolution at \(Q_{\beta \beta }\).

2 The XENON1T experiment

The XENON1T detector is a dual-phase xenon TPC which consists of a \(97~\mathrm {cm}\) length and \(96~\mathrm {cm}\) diameter cylindrical active detection volume containing 2 t of ultra-pure liquid xenon (LXe) out of a total of 3.2 t in the detector. Two arrays of Hamamatsu R11410-21 3\(''\) photomultiplier tubes (PMTs) [6] are arranged above and below the sensitive volume of the TPC. The side walls of the cylindrical volume are PTFE reflectors that enhance the light collection efficiency. Energy depositions from interactions in the LXe produce both scintillation photons and ionization electrons. The scintillation light signal (S1) is promptly detected by the PMTs. A grounded electrode, the gate, placed just \(\sim \) 2.5 mm below the liquid–gas interface, and a cathode placed at the bottom of the TPC produce an average electric field of 81 V/cm to drift electrons produced in the liquid upwards with a drift velocity of \(1.33\,\hbox {mm}/\upmu \hbox {s}\). An anode is placed 5 mm above the gate and the 8.1 kV/cm electric field between them extracts electrons into the gaseous xenon with an electron extraction efficiency calculated to be 96 % [7]. Here the electrons produce proportional scintillation light signal (S2), which is also detected by the PMTs. The time delay between S1 and S2 is used to reconstruct the interaction depth (z position) with a resolution down to 0.5 mm. The distribution of the S2 light on the top PMT array is used to reconstruct the xy position, reaching a resolution of 8 mm for S2 values above \(10^{3}\) photo-electrons (PE) [8]. The PMTs have an average quantum efficiency of 34.5% and channel-dependent gains of (1.0–5.0) \(\times 10^6\)  [9]. The signals are guided to Phillips 776 amplifiers that provide an additional amplification factor of 10. The output of the amplifiers is sent to CAEN V1724 waveform digitizer modules to record the signals at a sampling rate of 100 MHz with a 2.25 V dynamic range, a 40 MHz input bandwidth, and 14-bit resolution. The data acquisition system is described in detail in [10].

3 Signal reconstruction techniques

The data processing in XENON1T is performed with the modular software package Processor for Analyzing XENON (PAX) [8, 11]. This section describes several improvements to the low-level signal reconstruction routines of PAX for the dark matter search in order to optimize detector performance up to the MeV energy range.

3.1 Waveform saturation correction

XENON1T, designed for dark matter searches, features a signal readout system optimized to amplify and detect tiny signals down to single PE from individual PMTs  [10]. For interactions with energies \(\sim \) 1 MeV, several components, including the PMT voltage divider circuits, the amplifiers and the digitizers will saturate, resulting in distorted output S2 signals. A correction for saturation effects is thus critical for reconstructing signals at MeV energies with sufficient energy resolution for \(0\nu \beta \beta \) searches. The digitizer saturation occurs at energies above \(\sim 200\,\hbox {keV}\), corresponding to a total S2 signal on the order of \(10^{5}\,\hbox {PE}\) and as large as \(\sim 5\times 10^{4}\, \hbox {PE}\) in the PMT with the largest signal. The exact energy saturating the digitizers varies according to the location of the interaction. Such signals exceed the 2.25 V dynamic range of the digitizers and result in truncated waveforms (WFs). Non-linear responses of the PMT voltage divider circuits and the amplifiers are expected to occur at a higher energy of \(\sim 1\, \hbox {MeV}\), corresponding to an S2 signal on the order of \(10^{6}\, \hbox {PE}\). For these events, the analog (or the pre-digitizer) signals are distorted and no longer proportional to the number of initial photons [12]. Examples of S2 signals corresponding to those two cases are shown in Fig. 1.

Fig. 1
figure 1

Examples of saturated WFs from two S2s with a size of about \(2\times 10^{5}\, \hbox {PE}\) (top) and \(10^{6}\, \hbox {PE}\) (bottom). Each panel shows a WF (black) in one channel centred to time zero. Both WFs are truncated due to the range of the digitizer. The WF model, obtained from the sum of non-saturated WFs, is scaled and overlaid in the plot (red). The red shaded region each covers \(1\, \upmu \hbox {s}\) before the first truncated sample and used as a reference region, while the hatched region from the first truncated sample to the end of the pulse covers the range where WFs are corrected as the scaled WF model

The correction method described in this section is based on the temporal and spatial characteristics of the S2. The S2 has a wide (at least \(0.5\,\upmu \hbox {s}\) at \(1 \sigma \) width) and nearly identical temporal distribution across all channels because the proportional scintillation light is produced for the duration of electrons drifting from the liquid–gas interface to the anode. Additionally, this light is produced \(\simeq 7 \hbox {cm}\) below the top PMT array, with the majority of it hitting a few PMTs. Some PMTs, especially those on the top array and away from the xy coordinate of the S2, remain unsaturated. The pulse shape of S2 signals in those non-saturated channels are used to correct signals in the saturated channels. The correction procedure applies to individual peaks and is as follows:

  1. 1.

    Sorting all S2 WFs into two classes: saturated and non-saturated, based on whether the WF reaches the limit of the dynamic range of the digitizers.

  2. 2.

    All non-saturated WFs are summed together to get a WF model denoted as \({\text {W}}_{{\text {M}}}\). This WF model is an unbiased estimate of the S2 WF shape.

  3. 3.

    For each saturated WF denoted as \({\text {W}}_{{\text {S}}}\), the region before the first saturated sample is used as a reference region. We denote the integral of \({\text {W}}_{{\text {S}}}\) and of \({\text {W}}_{{\text {M}}}\) over the reference region as \({\text {A}}^{{\text {ref}}}_{{\text {S}}}\) and \({\text {A}}^{{\text {ref}}}_{{\text {M}}}\), respectively.

  4. 4.

    Each saturated WF is corrected as \({\text {A}}^{{\text {ref}}}_{{\text {S}}}/ {\text {A}}^{{\text {ref}}}_{{\text {M}}}\times {\text {W}}_{{\text {M}}}\) after the reference region.

  5. 5.

    The correction is applied to the region from the first saturated sample to the last sample of the WFs, \(1\,\upmu \hbox {s}\) after the WFs fall below the channel-specific trigger thresholds. In rare cases when another peak starts before the end of the WFs, the corrections stop right before the second peak.

Two representative examples of S2 each with a WF in a saturated channel are shown with the scaled model \({\text {W}}_{{\text {M}}}\) overlaid in Fig. 1. For the \(\text {S2} \sim 2\times 10^{5}\, \hbox {PE}\) shown in the top panel, the analog signal is not distorted and the falling edge of the \({\text {W}}_{{\text {S}}}\) agrees well with \({\text {W}}_{{\text {M}}}\). This is not the case with a larger \(\text {S2} \sim 10^{6}\,\hbox {PE}\), as shown in the bottom panel. Here, \({\text {W}}_{{\text {M}}}\) does not match the falling edge of the \({\text {W}}_{{\text {S}}}\). In particular, the undershoot of \({\text {W}}_{{\text {S}}}\) is not caused by the saturation of the digitizers but by saturation of the PMTs or amplifiers. However, the \({\text {W}}_{{\text {M}}}\) is still a valid model because the signal in the reference region is not yet large enough to induce saturation in those two components. The overshoot present on the right side is instead mostly due to secondary signals, as it will be clarified in Sect. 3.2. In order to rectify all saturation effects, the correction is extended to the last sample of \({\text {W}}_{{\text {S}}}\) in all cases. In addition to the impact on the energy reconstruction, the saturation correction also notably affects the position reconstruction and thus the spatial correction for the S1 and S2 signals, as shown in Sect. 3.3.

Unlike the S2, the S1 light is more evenly distributed among all PMTs and it is not amplified in the gas region. As a result, S1 signals from electronic recoils have negligible saturation, even for events with energies in MeV region. In addition to this, the scintillation photons are produced on much shorter timescales as in the S2 case, and building a WF model for S1 using non-saturated channels requires alignment of signals in all channels better than \(0.01\, \upmu \hbox {s}\). This is not achieved in XENON1T as the arrival time of each photon, the PMT time responses, and the length of readout cables are all different. For these reasons, the saturation correction described above is not applied to the S1.

3.2 Identification of primary and secondary signals

Secondary signals are defined as signals not directly caused by particle interactions in the LXe. They are associated with light and electron emission induced by S1s or S2s. Depending on the location of the emission we subdivide them into two main types. Gas present in PMTs can be ionized by accelerated electrons between the photocathode and the first dynode [9], producing after-pulse (AP) signals. Both photo-detachment of electronegative impurities and the photoelectric effect at the metal surfaces of the gate electrode produce electrons within the LXe, that in turn produce spurious S2 signals that we call photoionization (PI) signals [13].

Fig. 2
figure 2

A sample WF (after correction) of a high-energy event induced by a series of Compton scatters in the LXe. The summed WFs are shown as grey lines while the smoothed summed WFs are shown as the overlaid red lines in the insets. The WF of such an event typically has a narrow S1 peak and a few S2 peaks, each of which is followed by secondary signals from AP and PI processes. The effect of the algorithms on each peak is highlighted by the insets, with the final peak edges shown by vertical lines. The red points represent the local minima that define the end of the S1 signal and the start of each S2 signal. The blue points represent the threshold of 0.13% of the peak size. While secondary signals are clearly separated from the S1 peak, they overlap with S2 peaks

Since both AP and PI signals start to appear shortly (\(\simeq 1\, \upmu \hbox {s}\)) after the primary S1 or S2, they have significant effects on finding the peak boundaries. This leads to sizeable non-linearity and fluctuations in the reconstructed energy. Figure 2 shows the S1-S2 signal from a gamma-ray Compton scattering in the LXe after the saturation correction of Sect. 3.1 is applied. Each S1 and S2 is succeeded by AP and PI. While one can isolate the S1 from secondary signals based on the waveform, the S2s are too wide to separate such secondary signals out. Two algorithms are designed to discriminate and reject those secondary signals, as well as to identify individual interaction sites, using a WF summed over all channels. The two algorithms are complementary to each other. When they suggest splitting at two nearby points instead of the same point, the one closer to the primary signal is chosen as the final peak boundary.

  1. 1.

    To minimize the impact of noise, the summed WF (grey lines in Fig. 2) is smoothed (red lines) using a locally weighted smoothing method as in [14]. Local minima found in the smoothed summed WF are used to define peak boundaries marked as red points in Fig. 2. One of them is found in the gap between the S1 and secondary signals defining the end of the S1; two are found at the beginning of the S2 signals to split them from preceding secondary signals; the last one is found between overlapping S2 signals from two interaction sites.

  2. 2.

    A cutoff on the amplitude is set for each peak to define the extent of its falling edge. The cutoff threshold is placed at the value of a Gaussian function \(3-\sigma \) away from its center, with the height of the Gaussian matching the height of the peak. When the falling edge of the peak falls below this threshold, the peak is truncated in order to detach the tails from AP and PI. Thus, only 0.13% of the peak area is removed if the peak is Gaussian, as expected from the longitudinal diffusion of the electron cloud [15]. Marked as blue points in Fig. 2, the cutoff of the S1 is found to coincide with a local minimum; the cutoff points of the S2s split away most of the secondary signals, and their integrated area before the cutoff is approximately proportional to the size of S2.

3.3 Position reconstruction

The ability to reconstruct the three-dimensional position of events is a key advantage of dual-phase TPCs. The horizontal coordinates, xy, are reconstructed from the S2 light pattern in the top PMT array. Thus, to obtain an unbiased position, the WF correction is applied to the S2 signal. Calibration data from an external \(^{228} \hbox {Th}\) source are used to check the improvement of the position reconstruction induced by the saturation correction. The calibration source is placed at the side of the detector, close to the top of the TPC, which increases the number of saturated events and avoids the field distortion effect as in [8]. The radial position distribution of events from the \(^{208} \hbox {Tl}\) line at 2614.5 keV, mainly at the edge of the detector, is shown in Fig. 3. The same position reconstruction method is used, with (red) and without (blue) the saturation correction applied. The distribution of saturation-corrected reconstructed positions shows good agreement with the 48 cm maximum radius determined by the inner surface of the PTFE reflector, while the distribution without correction shows a significant inward bias.

Fig. 3
figure 3

Comparison of the radial position distribution of \(^{208} \hbox {Tl}\) events from external \(^{228} \hbox {Th}\) calibration, between data processed with (red) and without (blue) WF correction. Also shown here are the maximum radius of the TPC (black) and the distribution of simulated \(^{208} \hbox {Tl}\) events (green), scaled to fit the red distribution

Fig. 4
figure 4

Single-site (top) and multi-site (bottom) background event distributions in corrected S1 (cS1), and corrected S2 bottom (\({\hbox {cS2}}_b\)), space. Mono-energetic photo-absorption peaks of gamma-rays are labelled with their energies and corresponding sources. SS events with energies around \({\text {Q}}_{\beta \beta }\) are blinded

Fig. 5
figure 5

Anti-correlation between the measured light yield (LY) and charge yield (QY) using mono-energetic lines. These data points are from SS (blue) and MS (red) events in the inner 1 t fiducial volume. The values are different for SS and MS events at a given energy due to the energy-dependent ion-electron recombination processes. The best linear fit for combined SS and MS data points is overlaid as a solid line (grey)

Similar to the method detailed in [8], a feed-forward neural network is used to reconstruct xy coordinates. To improve the precision of the position reconstruction, a deeper network with four hidden layers is constructed using the Keras  [16] package with the TensorFlow  [17] backend. The dropout  [18] technique is applied to avoid over-fitting the network to the training set. Compared to [8], this neural network improves the position reconstruction precision by \(\simeq 30\%\) and leads to a more uniform response across the detector. Additionally, distortions in the position distribution due to an imperfect drift field are taken into account using the approach presented in [8].

4 Electronic recoil energy reconstruction

The energy resolution, which is particularly important for the \(0\nu \beta \beta \)-decay sensitivity, can be improved by applying the reconstruction techniques described in the previous sections. In this section, the calculation of the energy resolution using background data is described for single-site (SS) and for multi-site (MS) interactions.

4.1 Single and multi-site interactions

The number of interaction sites of an event is a key feature for discriminating background in the search for rare events. SS interactions encompass potential signals from rare physics processes like dark matter, \(0\nu \beta \beta \) and \(2\nu \beta \beta \) decays. In the latter two cases, the two betas are emitted at the same interaction point. Their penetration length in LXe is less than 3 mm  [19], indistinguishable with the spatial resolutions of the XENON1T detector. The expected signature is then a single scatter made of two coinciding, unresolved betas. Background contributions for these searches originate from interactions due to beta decays and gamma-rays. MS interactions, mainly due to multiple Compton scatters of gamma-rays (or the coincidence of two gamma-rays happening at the same time), are used to identify and constrain the background components.

Fig. 6
figure 6

Top: electronic recoil energy spectra of single-site (blue) and multi-site (red) events in the central 1t fiducial volume of XENON1T. SS events with energies around \({\text {Q}}_{\beta \beta }\) are blinded for the search for \(0\nu \beta \beta \) decay. The corresponding decaying isotope for the most visible peaks is labelled with a dashed vertical line. The MS spectrum has a lower rate at low energies due to the fiducial volume selection. Middle: the measured energy resolution for SS and MS events. The SS and MS resolutions as a function of energy are fit with \(a/\sqrt{E}+b\) and shown by the blue and red lines, respectively, while the shaded regions cover \(1-\sigma \) statistical uncertainty of the fits. The extrapolated values for the SS are \(a = (31.3 \pm 0.7)\) and \(b = (0.17 \pm 0.02)\). The resolution of XENON100  [20], LUX  [21], PandaX-II  [22] and EXO-200  [23] are also reported. Bottom: the relative energy shift from the true values for SS and MS events

4.2 Combined energy from S1 and S2

A linear, electric field independent relationship between energy and total number of produced quanta (either scintillation photons or ionization electrons) has been established in LXe dual-phase TPCs built for dark matter searches, such as XENON100  [20], LUX  [21], PandaX-II  [22], as well as LXe TPCs built for \(0\nu \beta \beta \), such as EXO-200  [23]. The energy transferred in an interaction can be expressed as

$$\begin{aligned} E = (n_{{\text {ph}}} + n_{{\text {e}}})\cdot W = \left( \frac{{\text {S1}}}{g_1} + \frac{{\text {S2}}}{g_2}\right) \cdot W \text { ,} \end{aligned}$$
(2)

where \(W = (13.7 \pm 0.2)\;\hbox {eV/quantum}\)   [24] is the average energy needed to produce either scintillation or ionization, and \(n_{{\text {ph}}}\) and \(n_{{\text {e}}}\) are the number of emitted photons and electrons. The scintillation photons and ionization electrons are then detected as S1 and S2 signals, with a photon detection efficiency of \(g_1\) and charge amplification factor of \(g_2\). These are detector-dependent parameters that are determined using mono-energetic peaks, including \(^{83\text {m}} \hbox {Kr}\), \(^{129\text {m}} \hbox {Xe}\), \(^{131\text {m}} \hbox {Xe}\), \(^{60} \hbox {Co}\) and \(^{208} \hbox {Tl}\). We rewrite Eq. (2) as

$$\begin{aligned} \text {QY} = -\frac{g_2}{g_1}\text {LY} + \frac{g_2}{W} \text { ,} \end{aligned}$$
(3)

where \(\hbox {QY} = {\text {S2}}/E\) and \(\hbox {LY} = {\text {S1}}/E\) are the mean charge yields and the mean light yields at each energy.

Figure 4 shows the distributions of background events for SS (top) and MS (bottom) interactions. The top PMT array is excluded from the summed S2 size to avoid detection efficiency changing suddenly in the xy plane under the non-operational PMTs. PMTs on the bottom array with large AP rate are also excluded. Leaving those PMTs out doesn’t increase the associated statistical fluctuations thanks to the amplification in gaseous xenon. S1 and S2 signals are then corrected with the relative detection efficiencies at different positions derived from the \(^{83\text {m}} \hbox {Kr}\) calibration. The derivation is updated from the approach detailed in [8], considering the electric field effect on the \(^{83\text {m}} \hbox {Kr}\) events. In particular, a linear correction depending on the depth of the interaction had to be added for both SS and MS events. This is slightly higher for the MS events due to a larger contribution of the AP and PI to the S2 signals. For a MS event, the combined S1 is corrected with the average of the relative light detection efficiencies at each of the S2s’ positions, and weighted by the size of the S2s.

The relative LY and QY are estimated by two-dimensional Gaussian fits to each monoenergetic peak above the background  [20, 25]. Figure 5 shows the relation between LY and QY. At given interaction energies, these measured values are different for SS and MS events due to the energy-dependent ion-electron recombination processes. \(g_2\) and \(g_1\) depend on the specific characteristics of the detector and on the type of interaction. As SS and MS light and charge yields from ER interactions provide consistent values, they are fitted together. The extrapolated parameters are

$$\begin{aligned} g_1&= (0.147 \pm 0.001) \, \text {PE/photon} \, , \end{aligned}$$
(4)
$$\begin{aligned} g_2&= (10.53 \pm 0.04) \, \text {PE/electron} \, , \end{aligned}$$
(5)

The reconstructed energy is then calculated using these values for both SS and MS events.

Fig. 7
figure 7

Fit on the \(^{214} \hbox {Bi}\) peaks above 2 MeV, \(\chi ^2/\hbox {d.o.f} = 113.7/118\). The extracted value and standard deviation for the peak at higher energy are \(\mu = 2204.4\,\hbox {keV}\) and \(\sigma = 17.9\, \hbox {keV}\), respectively. The computed resolution is then \(\sigma /\mu = 0.81\%\)

4.3 Linearity and resolution of the reconstructed energy

The reconstructed energy spectra for both SS and MS data are shown in the top panel of Fig. 6. Mono-energetic gamma lines from radioactive decays are fitted with Gaussian distributions above a background characterized by a constant or linear function around the peaks. An example is shown in Fig. 7. In other cases, when the background around the peak is rapidly changing, an exponential function is added to the fit as well. The fits yield the resolution of the reconstructed energy, \(\sigma (E_{{\text {r}}})/\mu (E_{{\text {r}}})\), and its shift from the nominal value, \((\mu (E_{{\text {r}}})-E_{{\text {t}}})/E_{{\text {t}}}\), the reconstructed energy being \(E_{{\text {r}}}\) when the true value is \(E_{{\text {t}}}\), with a mean value of \(\mu (E_{{\text {r}}})\) and a standard deviation of \(\sigma (E_{{\text {r}}})\). The shift observed across the entire energy range for both SS and MS data is \(\le 0.4 \%\). For comparison, the S2 signals on the bottom PMT array are biased up to \(-3\%\) at 2.5 MeV if the saturation correction is not applied. The excellent linearity of the energy response further ensures that the \(g_2\) and \(g_1\) calibrated at higher energy are applicable to low energy signals.

The energy resolution of SS data acquired during 246.7 days of dark matter search by XENON1T is (\(0.80 \pm 0.02\)) % in 1-t fiducial mass at 2.46 MeV, to be compared with the 4.2 % reported for the dual-phase LXe TPC of the PandaX-II experiment [22] and the energy resolution of (\(1.15 \pm 0.02\)) % achieved in EXO-200 [23]. The achieved resolution for MS events at 2.46 MeV is (\(0.90 \pm 0.03\))%. The slightly lower resolution from MS data with respect to SS data is due to limitations in the identification, reconstruction and corrections of both the S1 and S2.

5 Conclusions and outlook

We have presented signal reconstruction and correction methods designed to improve the energy linearity and resolution at MeV energies in the XENON1T dual-phase TPC. We have devised procedures to correct S2 signals with saturation due to both the digitizers’ dynamic range and distortions caused by the non-linear response of the PMT voltage divider circuits and the amplifiers. We obtained an unprecedented relative energy resolution of \(1\, \sigma /\mu = (0.80 \pm 0.02)\,\%\) at 2.46 MeV in a drift field of 81 V/cm. This resolution is mostly limited by fluctuations in the scintillation and ionization signals. The photon detection efficiency \(g_1\) determines the fluctuations in the scintillation signal. The mean electrons’ drift length before absorption by electronegative impurities in the liquid determines the fluctuations in the ionization signal. In XENON1T, the mean drift length is \(\ge 80\, \hbox {cm}\), leading to a \(\simeq 30\%\) survival probability of an ionization signal at the bottom region of the TPC. This is significantly higher than for the scintillation channel, where the efficiency is \(\simeq 12 \%\). Further improvements in energy resolution can be achieved with larger photosensor coverage and higher quantum efficiency which would reduce the fluctuations in the scintillation signal.

The upcoming XENONnT experiment, an upgrade of XENON1T with a larger TPC and reduced background, is expected to start taking data in 2020. Several detector improvements will enhance the energy reconstruction of high-energy events. Firstly, the dynamic range of the S2 signal will be extended. The amplifiers of the top PMTs will feature dual gains, a high-gain channel with 10X amplification, and a low-gain channel with a 2X attenuation. Secondly, smaller fluctuations in the ionization channel are expected due to a longer mean drift length of electrons before absorption, thanks to a cryogenic LXe purification system with higher circulation speed. Beside the hardware upgrades, the energy reconstruction in XENONnT will still benefit from the WF correction algorithm developed in this work, to address the distortions on the analog signals such as those due to the PMT voltage divider circuits. The resulting improvement in energy resolution and linearity, coupled with the expected lower background of the new detector, will make it well-suited to search for rare events beyond those expected from dark matter particles, such as the neutrinoless double-beta decay of \(^{136} \hbox {Xe}\).